THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection

Home Page:https://arxiv.org/abs/2405.14458

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

python方式model = YOLOv10('yolov10n.pt')报错_pickle.UnpicklingError: invalid load key, '\xe0'.

burning-star-x opened this issue · comments

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'.