search

Daniil-Osokin / lightweight-human-pose-estimation-3d-demo.pytorch

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.
Apache License 2.0
653 stars 137 forks source link

Poses without neck are not skipped #67

Closed platvlad closed 3 years ago

platvlad commented 3 years ago

Hi! Thank you for your work!

I explore the code and it seems that this line which should check that the neck was found, does not do anything. It checks x coordinate of the keypoint to be equal to -1, but coordinates are already divided by upsample_ratio few strings above. Seems it should check z coordinate of the keypoint. The same goes for this line

cuichangrui commented 3 years ago

Hello, I reported the following error while running the code：

Traceback (most recent call last): File "demo.py", line 59, in net = InferenceEnginePyTorch(args.model, args.device, use_tensorrt=args.use_tensorrt) File "/home/cui/lightweight-human-pose-estimation-3d-demo.pytorch/modules/inference_engine_pytorch.py", line 22, in init checkpoint = torch.load(checkpoint_path, map_location='cpu') File "/home/cui/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 579, in load with _open_file_like(f, 'rb') as opened_file: File "/home/cui/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 230, in _open_file_like return _open_file(name_or_buffer, mode) File "/home/cui/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 211, in init super(_open_file, self).init(open(name, mode)) FileNotFoundError: [Errno 2] No such file or directory: 'human-pose-estimation-3d.pth'

May I ask how to solve this？

platvlad commented 3 years ago

@cuichangrui It just can not find the file 'human-pose-estimation-3d.pth'. Check that the directory from which you run demo.py contains this file. Or just provide the full path to this file

cuichangrui commented 3 years ago

Thank you for your reply.Do you know where I can find the file 'human-pose-estimation-3d.pth'?I don't know where this file is

platvlad commented 3 years ago

@cuichangrui It is a pretrained model which is available at Google Drive. See README

Daniil-Osokin commented 3 years ago

@platvlad Hi, Vladislav! Thanks, looks so. Need some time to check it.

cuichangrui commented 3 years ago

@cuichangrui 这是一个预先训练的模型，可以在Google云端硬盘上使用。请参阅自述文件

Thank you for your reply. I was wondering how do you train pictures, not with a camera

Daniil-Osokin commented 3 years ago

@cuichangrui Will #17 help?

cuichangrui commented 3 years ago

@cuichangrui Will #17 help? Thank you for your reply.Have a nice day