MMDetection3.0以上推理
mmdetection3.02以后推理代码有所更改,具体代码为:
from mmdet.apis import inference_detector, init_detector #, show_result_pyplot
from mmdet.apis import DetInferencer
def main():
# config文件
config_file = 'work_dirs/retinanet-TinyCityPerson/retinanet_r50_fpn_1x_coco.py'
# 训练好的模型
checkpoint_file = 'work_dirs/retinanet-TinyCityPerson/epoch_12.pth'
#初始化推理器
inferencer = DetInferencer(model = config_file, weights = checkpoint_file)
imgs_path = "sssssss"#这里放图片路径
out_dir = “sss”#这里放输出路径
inferencer(imgs_path , out_dir=out_dir )
if __name__== '__main__':
main()
如需更改画框颜色和删除类别名,需要在 ./mmdet/visualizetion/local_visualizer.py中更改画框颜色和类别名显示。