Skip to content

Latest commit

 

History

History
55 lines (26 loc) · 3.62 KB

terminal commands.md

File metadata and controls

55 lines (26 loc) · 3.62 KB

Following are the terminal commands used in this study with the config files

To run this, please add the corresponding config files in the config/ directory in ViTPose repository.

The checkpoints could be found at: https://www.dropbox.com/scl/fo/cs5bdynig2a8jazpomul7/AFJ_f9mKHxlsrR60eGn0SRU?rlkey=tw1y19spyxtsca94pz6on2z1j&st=uduvogbm&dl=0

Train

bash tools/dist_train.sh configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192.py 4 --cfg-options model.pretrained=weights/small_pretrained.pth --seed 0

Train from checkpoint

bash tools/dist_train.sh configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192.py 4 --cfg-options resume_from=work_dirs/ViTPose_small_hi5_all_256x192/epoch_50.pth model.pretrained=weights/small_pretrained.pth --seed 0

Test

bash tools/dist_test.sh ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192.py ViTPose/work_dir/ViTPose_small_hi5_all_256x192_500k/best_AUC_epoch_48.pth 4

Test Hi5-L

bash tools/dist_test.sh /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_hi5_all_256x192_500k/epoch_49.pth 4

Test Hi5-M

bash tools/dist_test.sh /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_hi5_all_256x192_100k/best_AUC_epoch_125.pth 4

Test Hi5-S

bash tools/dist_test.sh /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_hi5_all_256x192_10k/best_AUC_epoch_179.pth 4

Test OneHand10k

bash tools/dist_test.sh /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/ViTPose_small_onehand10k_all_256x182.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_onehand10k_all_256x182/best_AUC_epoch_399.pth 4

Test OneHand10k on Mediapipe, 11k

bash tools/dist_test.sh /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/ViTPose_small_onehand10k_all_256x182_11k.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_onehand10k_all_256x182/best_AUC_epoch_399.pth 4

Test Hi5L on Mediapipe

bash tools/dist_test.sh /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192_11k.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_hi5_all_256x192_500k/epoch_49.pth 4

Demo

OneHand10k

python demo/top_down_img_demo.py /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/ViTPose_small_onehand10k_all_256x182.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_onehand10k_all_256x182/best_AUC_epoch_399.pth --img-root /scratch/mhasan9/hi5/data/onehand10k/ --json-file /scratch/mhasan9/hi5/data/onehand10k/annotations/onehand10k_test.json --out-img-root vis_results_onehand10k

Hi5-L

python demo/top_down_img_demo.py /scratch/mhasan9/hi5/ViTPose/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/hi5/ViTPose_small_hi5_all_256x192.py /scratch/mhasan9/hi5/ViTPose/work_dirs/ViTPose_small_hi5_all_256x192_500k/epoch_49.pth --img-root /scratch/mhasan9/hi5/data/onehand10k/ --json-file /scratch/mhasan9/hi5/data/onehand10k/annotations/onehand10k_test.json --out-img-root vis_results_hi5