Traceback (most recent call last):
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\main.py", line 110, in
main(args.img_path)
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\main.py", line 82, in main
saliency_method = YOLOV5GradCAM(model=model, layer_name=args.target_layer, img_size=input_size)
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\models\gradcam.py", line 44, in __init__
self.model(torch.zeros(1, 3, *img_size, device=device))
File "E:\anaconda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "G:\new_python_project\Gard-CAM热力图可视化\yolov5-6.1-gradcam\models\yolo_v5_object_detector.py", line 167, in forward
prediction, logits, _ = self.model(img, augment=False)
ValueError: not enough values to unpack (expected 3, got 2)
为啥我运行就报这个错误了呢?还是拿老师的原项目运行的