_pickle.UnpicklingError: invalid load key, 'v'. 错误

724852499 commented 1 month ago

root@ca9782e90bf9:/home/pdf/PDF-Extract-Kit# python pdf_extract.py --pdf /home/pdf/test.pdf Namespace(pdf='/home/pdf/test.pdf', output='output', vis=False, render=False) 2024-08-15 17:08:10 Started! Traceback (most recent call last): File "/home/pdf/PDF-Extract-Kit/pdf_extract.py", line 98, in mfd_model = mfd_model_init(model_configs['model_args']['mfd_weight']) File "/home/pdf/PDF-Extract-Kit/pdf_extract.py", line 31, in mfd_model_init mfd_model = YOLO(weight) File "/usr/local/lib/python3.10/site-packages/ultralytics/models/yolo/model.py", line 23, in init super().init(model=model, task=task, verbose=verbose) File "/usr/local/lib/python3.10/site-packages/ultralytics/engine/model.py", line 142, in init self._load(model, task=task) File "/usr/local/lib/python3.10/site-packages/ultralytics/engine/model.py", line 294, in _load self.model, self.ckpt = attempt_load_one_weight(weights) File "/usr/local/lib/python3.10/site-packages/ultralytics/nn/tasks.py", line 855, in attempt_load_one_weight ckpt, weight = torch_safe_load(weight) # load ckpt File "/usr/local/lib/python3.10/site-packages/ultralytics/nn/tasks.py", line 781, in torch_safe_load ckpt = torch.load(file, map_location="cpu") File "/usr/local/lib/python3.10/site-packages/ultralytics/utils/patches.py", line 86, in torch_load return _torch_load(*args, kwargs) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1040, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1262, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, 'v'. 执行命令报错，什么原因呢