一、下载MMDETECTION官方源码
二、找到benchmark.py
找打def parse_args():
函数:
根据需求修改相应参数。
执行如下命令:
第一个参数为训练的模型py文件,第二个参数为训练后的模型权重pth文件,
python tools/analysis_tools/benchmark.py work_dirs/faster-rcnn/faster-rcnn_r50_fpn_1x_coco.py --checkpoint work_dirs/faster-rcnn/epoch_190.pth
运行结果如下:
即可得到FPS和推理时间