python方式model = YOLOv10('yolov10n.pt')报错_pickle.UnpicklingError: invalid load key, '\xe0'.
burning-star-x opened this issue · comments
burning-star-x commented
from ultralytics import YOLOv10
# model = YOLOv10('yolov10n_gg_nc26.yaml').load('yolov10n.pt')
# model = YOLOv10('yolov10n.yaml').load('yolov10n.pt')
# model = YOLOv10('yolov10s.yaml').load('yolov10s.pt')
# model = YOLOv10('yolov10n.pt')
model = YOLOv10('yolov10n.pt')
# Train the model
results = model.train(data='gg.yaml',epochs=100, imgsz=640, batch=1, optimizer='SGD', lr0=0.01, device=0)
报错如下
Traceback (most recent call last):
File "D:\AllMyProjects\AllIDEs\Python\000_PythonProjects\yolov10\my_train.py", line 7, in <module>
model = YOLOv10('yolov10n.pt')
File "D:\AllMyProjects\AllIDEs\Python\000_PythonProjects\yolov10\ultralytics\models\yolov10\model.py", line 14, in __init__
super().__init__(model=model, task=task, verbose=verbose)
File "D:\AllMyProjects\AllIDEs\Python\000_PythonProjects\yolov10\ultralytics\engine\model.py", line 141, in __init__
self._load(model, task=task)
File "D:\AllMyProjects\AllIDEs\Python\000_PythonProjects\yolov10\ultralytics\engine\model.py", line 230, in _load
self.model, self.ckpt = attempt_load_one_weight(weights)
File "D:\AllMyProjects\AllIDEs\Python\000_PythonProjects\yolov10\ultralytics\nn\tasks.py", line 803, in attempt_load_one_weight
ckpt, weight = torch_safe_load(weight) # load ckpt
File "D:\AllMyProjects\AllIDEs\Python\000_PythonProjects\yolov10\ultralytics\nn\tasks.py", line 729, in torch_safe_load
ckpt = torch.load(file, map_location="cpu")
File "C:\Users\filly\.conda\envs\yqqsqz_yolov10\lib\site-packages\torch\serialization.py", line 815, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "C:\Users\filly\.conda\envs\yqqsqz_yolov10\lib\site-packages\torch\serialization.py", line 1033, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '\xe0'.
torch版本是conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
其他东西都用requirments.txt装的
app.py能正常跑
这句命令能正常跑yolo detect train data=gg.yaml model=yolov10n.yaml epochs=500 batch=256 imgsz=640 device=0
但是用脚本方式就报错_pickle.UnpicklingError: invalid load key, '\xe0'.