此博客介绍了如何在PaddleDetection的deploy/python/infer.py文件中添加批量预测功能，并允许用户指定特定类别（如person）的预测结果保存。作者对predict_image函数进行了扩展，新增了--infer_dir参数，以便处理目录下的多张图片。同时，对visualize函数进行了优化，可以根据类别区分保存路径。
PaddleDetection/deploy/python/infer.py导出模型的推理预测程序中没有批量预测和指定目标保存。现对代码进行修改，增添功能。
在infer.py 中predict_image函数中进行修改，进行批量测试
加入超参数–infer_dir
在main函数中进行修改，增加参数
#
parser.add_argument(
"--infer_dir", type=str, default='', help="Directory for images to perform inference on.")
#
最低0.47元/天 解锁文章
扫一扫
3万+
到【灌水乐园】发言
