Open Riverzxz opened 1 year ago
Evaluation bash tools/dist_test.sh $CONFIG $CHECKPOINT $NUM_GPU --eval keypoints ex) CUDA_VISIBLE_DEVICES=1,2 bash tools/dist_test.sh configs/petr/petr_r50_16x2_100e_coco.py checkpoint/petr_r50_16x2_100e_coco.pth 2 --eval keypoints
Training bash tools/dist_train.sh $CONFIG $NUM_GPU ex) CUDA_VISIBLE_DEVICES=1,2 bash tools/dist_train.sh configs/petr/petr_r50_16x2_100e_coco.py 2
Inference ex) python tools/test.py configs/petr/petr_r50_16x2_100e_coco.py checkpoint/petr_r50_16x2_100e_coco.pth --show-dir ./results
Sorry to ask you again.I have an error
Traceback (most recent call last):
File "tools/test.py", line 251, in
Did you download coco datasets and annotation files? you should download annotation files and images from [https://cocodataset.org/#download]. and fix data root on 'data_root = '/dataset/public/coco/', which is in [configs/base/datasets/coco_keypoint.py].
Thank you for your help! !I am currently doing an Inference of the coco dataset. If I want to use a model to visually test a video, such as the dekr (cvpr2021) model: // python tools/inference_demo.py --cfg experiments/coco/inference_demo_coco.yaml \ --videoFile ../multi_people.mp4 \ --outputDir output \ --visthre 0.3 \ TEST.MODEL_FILE model/pose_coco/pose_dekr_hrnetw32.pth // What do I need to do in petr model?
I couldn't find the video inference code in this repository.. I recommend converting mp4 videos into png frames for inference... or fix codes for video. I am sorry that I cannot help you.
I use following code to evaluate the PETR ,but I have an error.
code:CUDA_VISIBLE_DEVICES=1 bash tools/dist_test.sh configs/petr/petr_r50_16x2_100e_coco.py checkpoint/petr_r50_16x2_100e_coco.pth 1 --show-dir /output --eval keypoints
could you please help me?
It seems like torch vision's problem, please use following configuration: python 3.7 + torch 1.8 + cuda 10.1 + mmcv 1.5.3 + mmdet 2.25
Thanks for your work, I am in the process of using petr. How to start the training, testing and demo testing of the petr multi-person pose model? Can the steps be given? Looking forward to your reply and answer