Giters

mit-han-lab / litepose

[CVPR'22] Lite Pose: Efficient Architecture Design for 2D Human Pose Estimation

Home Page:https://hanlab.mit.edu

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

help

731076467 opened this issue · comments

commented

Hello, I have tested the coco test image, but what should I do if I want to test my own image and want to get its key points to see the effect?

commented

I wrote a quick addition to test a directory with custom images, you can find it here: https://github.com/random9v2/litepose/tree/test_img_dir

You will have to install natsort via conda/pip. For running the code, choose your image directory and run it, e.g., with:

python valid.py --cfg experiments/crowd_pose/mobilenet/mobile.yaml --superconfig mobile_configs/search-XS.json TEST.MODEL_FILE checkpoints/LitePose-Auto-XS_crowd.pth.tar DATASET.DATASET_TEST image_dir DATASET.ROOT dataset/my_img/ PRINT_FREQ 1
commented

I wrote a quick addition to test a directory with custom images, you can find it here: https://github.com/random9v2/litepose/tree/test_img_dir

You will have to install natsort via conda/pip. For running the code, choose your image directory and run it, e.g., with:

python valid.py --cfg experiments/crowd_pose/mobilenet/mobile.yaml --superconfig mobile_configs/search-XS.json TEST.MODEL_FILE checkpoints/LitePose-Auto-XS_crowd.pth.tar DATASET.DATASET_TEST image_dir DATASET.ROOT dataset/my_img/ PRINT_FREQ 1

Although I wrote one myself, thank you very much for your help!

commented

I wrote a quick addition to test a directory with custom images, you can find it here: https://github.com/random9v2/litepose/tree/test_img_dir
You will have to install natsort via conda/pip. For running the code, choose your image directory and run it, e.g., with:

python valid.py --cfg experiments/crowd_pose/mobilenet/mobile.yaml --superconfig mobile_configs/search-XS.json TEST.MODEL_FILE checkpoints/LitePose-Auto-XS_crowd.pth.tar DATASET.DATASET_TEST image_dir DATASET.ROOT dataset/my_img/ PRINT_FREQ 1

Although I wrote one myself, thank you very much for your help!

I have the same problem and if I want to test my own image. Do you have the correct inference code? thanks