YOLO训练模型时遇到这种问题怎么修改(标签-python|关键词-File)

(yolov5) PS D:\yolov5> python train.py --cfg models/yolov5x-train.yaml --data data/tarin.yaml --weights yolov5x.pt --epoch 150 --batch-size 32

Traceback (most recent call last):
File "train.py", line 51, in
from utils.loggers import Loggers
File "D:\yolov5\utils\loggers_init_.py", line 14, in
from utils.loggers.clearml.clearml_utils import ClearmlLogger
File "D:\yolov5\utils\loggers\clearml\clearml_utils.py", line 8, in
from torchvision.utils import draw_bounding_boxes
ImportError: cannot import name 'draw_bounding_boxes' from 'torchvision.utils' (D:\anadas\envs\yolov5\lib\site-packages\torchvision\utils.py)

可能是你的torchvision包版本不对，或者模块损坏
你可以尝试重新安装torchvision包

torch和torchvision版本可能不一致。