如果想将预测结果保存为特定的文件,用于离线评估,目前 MMYOLO 支持 json 和 pkl 两种格式。
json 文件仅保存 `image_id`、`bbox`、`score` 和 `category_id`; json 文件可以使用 json 库读取。
pkl 保存内容比 json 文件更多,还会保存预测图片的文件名和尺寸等一系列信息; pkl 文件可以使用 pickle 库读取。
如果想将预测结果输出为 json 文件,则命令如下:
python tools/test.py ${CONFIG} ${CHECKPOINT} --json-prefix ${JSON_PREFIX}
--json-prefix
后的参数输入为文件名前缀(无需输入 .json
后缀),也可以包含路径。举一个具体例子:
python tools/test.py configs\yolov5\yolov5_s-v61_syncbn_8xb16-300e_coco.py yolov5_s-v61_syncbn_fast_8xb16-300e_coco_20220918_084700-86e02187.pth --json-prefix work_dirs/demo/json_demo
运行以上命令会在 work_dirs/demo
文件夹下,输出 json_demo.bbox.json
文件。
如果想将预测结果输出为 pkl 文件,则命令如下:
python tools/test.py ${CONFIG} ${CHECKPOINT} --out ${OUTPUT_FILE} [--cfg-options ${OPTIONS [OPTIONS...]}]
--out
后的参数输入为完整文件名(必须输入 .pkl
或 .pickle
后缀),也可以包含路径。举一个具体例子:
python tools/test.py configs\yolov5\yolov5_s-v61_syncbn_8xb16-300e_coco.py yolov5_s-v61_syncbn_fast_8xb16-300e_coco_20220918_084700-86e02187.pth --out work_dirs/demo/pkl_demo.pkl
运行以上命令会在 work_dirs/demo
文件夹下,输出 pkl_demo.pkl
文件。