mmdetection pkl 指定输出位置

对于 MMDetection 框架，如果你想要将 `.pkl` 文件输出到指定的位置，可以使用以下方法：

首先，导入相关的库和模块：

from mmcv import Config
from mmdet.apis import inference_detector, init_detector, save_result_pyplot

然后，加载配置文件和权重：

config_file = 'path/to/config_file.py'
checkpoint_file = 'path/to/checkpoint_file.pth'

# 加载配置文件
cfg = Config.fromfile(config_file)
# 构建模型并加载权重
model = init_detector(config_file, checkpoint_file, device='cuda:0')

接下来，进行推理并保存结果：

img = 'path/to/input_image.jpg'  # 输入图片路径
output_file = 'path/to/output.pkl'  # 输出 .pkl 文件路径

# 进行推理
result = inference_detector(model, img)

# 保存结果为 .pkl 文件
model.module.save_results(result, output_file)

以上代码片段中，`config_file` 是模型的配置文件路径，`checkpoint_file` 是训练好的权重文件路径。`img` 是需要进行推理的输入图片路径，`output_file` 是保存输出结果的 `.pkl` 文件路径。

通过以上步骤，你可以将 MMDetection 框架的预测结果保存到指定位置。请根据实际情况修改文件路径和其他参数。