MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
http://mvig.org/research/alphapose.html
Other
7.83k stars 1.96k forks source link

scripts/demo_inference.py for Halpe datasets(26 keypoints) cannot work. #1171

Closed ForLoveandPeace closed 10 months ago

ForLoveandPeace commented 10 months ago

Hi, thanks for your work. I tried it with coco model(17 kps) and halpe model(26 kps) on Colab. it works well with coco model. But it cannot load YOLO model with halpe model.

Loading YOLO model.. Traceback (most recent call last): File "/content/AlphaPose/scripts/demo_inference.py", line 183, in pose_model.load_state_dict(torch.load(args.checkpoint, map_location=args.device)) File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 815, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1051, in _legacy_load typed_storage._untyped_storage._set_from_file( RuntimeError: unexpected EOF, expected 805352 more bytes. The file might be corrupted.

Sorry! That's my fault! The model is not downloaded completely.