基于YOLOV5 的多分类 + 关键点检测
小太阳&*_*&:
请问一下,为什么推理出现了这个错误File "C:\Users\Administrator\Downloads\yolov5_v6_plate_heading-master\detect.py", line 308, in
main(opt)
File "C:\Users\Administrator\Downloads\yolov5_v6_plate_heading-master\detect.py", line 303, in main
run(**vars(opt))
File "F:\miniconda3\envs\keypoint-6\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "C:\Users\Administrator\Downloads\yolov5_v6_plate_heading-master\detect.py", line 151, in run
pred = model(img, augment=augment, visualize=visualize)[0]
File "F:\miniconda3\envs\keypoint-6\lib\site-packages\torch\nn\modules\module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
TypeError: forward() got an unexpected keyword argument 'visualize'
|