search

mks0601 / TF-SimpleHumanPose

TensorFlow implementation of "Simple Baselines for Human Pose Estimation and Tracking", ECCV 2018
335 stars 90 forks source link

The code about test does not run successfully.......... #58

Closed YHDang closed 4 years ago

YHDang commented 4 years ago

Hi. I downloaded the human_detection.json, and ran the test.py by python test.py --gpu 0-1 --test_epoch 140. But the code couldn't run successfully. The errors are as follows. Traceback (most recent call last): File "test.py", line 256, in test(int(args.test_epoch)) File "test.py", line 232, in test d.evaluation(result, annot, cfg.result_dir, cfg.testset) File "/home/dyh/Sources/python/algorithm/TF-SimpleHumanPose-master/main/../data/MPII/dataset.py", line 90, in evaluation savemat(result_path, mdict=result) File "/home/dyh/anaconda3/envs/tensorflow/lib/python3.6/site-packages/scipy/io/matlab/mio.py", line 219, in savemat MW.put_variables(mdict) File "/home/dyh/anaconda3/envs/tensorflow/lib/python3.6/site-packages/scipy/io/matlab/mio5.py", line 834, in put_variables for name, var in mdict.items(): AttributeError: 'list' object has no attribute 'items'

How can I address this problem? I don't know if I downloaded the human_detection.json by mistakes. Could you provide the specific version?
Thank you~

mks0601 commented 4 years ago

The error comes from savemat from ../data/MPII/dataset.py. It seems the savemat only accept a dict, not a list. However, in the test.py, test_net function returns list. It seems you should convert the list to dict format. I think this is not very hard, so you can try? I'll fix this problem soon. Thanks for reporting!

YHDang commented 4 years ago

Yeah! Thanks for your reply. I can do it

mks0601 commented 4 years ago

Sorry, I only tested this code on PoseTrack and MSCOCO dataset, which are the most widely used datasets.

YHDang commented 4 years ago

OK, thanks. By the way, can I use the trained model to evaluate my own video ( I only need the visualization)?

mks0601 commented 4 years ago

Of course. You can find the download link at README. Here is the link. https://drive.google.com/drive/folders/1DAyOVI-8q-gh1ooe6cMx6RA16GJ03Efn?usp=sharing

YHDang commented 4 years ago

Thanks very much. And I have one question. Where can I find the use of the optical flow in the code?

------------------ 原始邮件 ------------------ 发件人: "Gyeongsik Moon"; 发送时间: 2020年1月2日(星期四) 中午12:49 收件人: "mks0601/TF-SimpleHumanPose"; 抄送: "YHDang"769731469@qq.com;"Author"; 主题: Re: [mks0601/TF-SimpleHumanPose] The code about test does not run successfully.......... (#58)

Of course. You can find the download link at README. Here is the link. https://drive.google.com/drive/folders/1DAyOVI-8q-gh1ooe6cMx6RA16GJ03Efn?usp=sharing

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

mks0601 commented 4 years ago

I did not implement the tracking part.