fromargparseimportargumentparserimportosfrommdet.apisimportinference _ detector，init_detector #， show _ result _ pyplotimportcv2def show _ result _ py plot (model，img，result，score_thr=0.3， fig_size=(15 ) 15 10 ) :“”“visualizethedetectionresultsontheimage.args :模型(nn.module ) 3360 theloadeded 0 imagefilenameorloadedimage.result (tuple [ list ] or list ) : The detection result，canbeeither ) bbox，原始设备3360 figuresizeofthepyplotfigure.' ' if hasattr (model，' module ' ) : model=model.module img=model.show _ re SSE show=false(returnimgdefmain ) : # config文件config _ file='/root/mm detection/work _ dirs/faster _ rcnn _ rcn 训练的模型check point _ file='/root/mm detection/work _ dirs/faster _ model=init _ detector (config _ file，chech ) check point _ fille device=' cuda :0 ' ) #检测到图像路径name='/root/mm detection/data/coco/val 2017/00088.jpg ' #后ral ' ifno tos.path.exists (out_dir ) :OS.mkdir ) out _ dir ) result=inference_detector ) model，name ) ime