Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

from mmpose.apis import init_model 报错 _pickle.UnpicklingError: invalid load key, '<'. #236

Closed
WJMismybestfriend opened this issue Dec 11, 2024 · 0 comments

Comments

@WJMismybestfriend
Copy link

WJMismybestfriend commented Dec 11, 2024

cuda:12.0
mmcv:2.1.0
mmpose:1.3.2
mmdet:3.2.0
torch:2.0.1+cu118
dw-ll_ucoco_384.pth文件没有损坏
执行:python -m scripts.realtime_inference --inference_config configs/inference/realtime.yaml --batch_size 4
报错:
please download ffmpeg-static and export to FFMPEG_PATH.
For example: export FFMPEG_PATH=/musetalk/ffmpeg-4.4-amd64-static
Loads checkpoint by local backend from path: ./models/dwpose/dw-ll_ucoco_384.pth
Traceback (most recent call last):
File "/usr/local/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/usr/local/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
File "/home/.../MuseTalk-main/scripts/realtime_inference.py", line 14, in
from musetalk.utils.preprocessing import get_landmark_and_bbox,read_imgs,coord_placeholder
File "/home/.../MuseTalk-main/musetalk/utils/preprocessing.py", line 19, in
model = init_model(config_file, checkpoint_file, device=device)
File "/home/.../MuseTalk-main/myenv/lib/python3.10/site-packages/mmpose/apis/inference.py", line 110, in init_model
ckpt = load_checkpoint(model, checkpoint, map_location='cpu')
File "/home/.../MuseTalk-main/myenv/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 636, in load_checkpoint
checkpoint = _load_checkpoint(filename, map_location, logger)
File "/home/.../MuseTalk-main/myenv/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 548, in _load_checkpoint
return CheckpointLoader.load_checkpoint(filename, map_location, logger)
File "/home/.../MuseTalk-main/myenv/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 330, in load_checkpoint
return checkpoint_loader(filename, map_location)
File "/home/.../MuseTalk-main/myenv/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 347, in load_from_local
checkpoint = torch.load(filename, map_location=map_location)
File "/home/.../MuseTalk-main/myenv/lib/python3.10/site-packages/torch/serialization.py", line 815, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/.../MuseTalk-main/myenv/lib/python3.10/site-packages/torch/serialization.py", line 1033, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '<'.

@WJMismybestfriend WJMismybestfriend changed the title 加载dwpose模型报错 _pickle.UnpicklingError: invalid load key, '<'. from mmpose.apis import init_model 报错 _pickle.UnpicklingError: invalid load key, '<'. Dec 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant