Lightweight-human-pose-estimation.pytorch: train No.3

Created on 21 Mar 2021  ·  6Comments  ·  Source: Daniil-Osokin/lightweight-human-pose-estimation.pytorch

Hello! I want to ask two questions. Firstly, I tried to train the No.3 (--from-mobilenet)and train 285000 iters. However, I did not detect any points and lines when I ran the Demo with it. Do you know what the problem is?Is it because I didn't train for step four or five? Secondly, Can you tell me how much you lost at the end of your training? I want to know the approximate loss when we can stop training. I trained 285000 iters, my losses are as follows:
image

All 6 comments

I ran into this problem during training, and I commented out this line of code in train.py(#evaluate(val_labels, val_output_name, val_images_folder, net)).Is this line of code commented out, so there is no result? Do you have any solutions?
image

Hello,I have seen your previous reply. Now I am making 19 key points and have modified the code. The data sets I used are from Val2017 and Train2017 downloaded from the official website of Coco. Do you mean that I need to re-label the data? Isn't the data set downloaded from the official website of Coco already marked?
image

Hi! First of all check if provided pre-trained model works for validation. This will say if validation works. Then check output file with prediction results from your checkpoint. If it is empty, then such error may occur. So, if it is empty, try to visualize the avg_heatmaps to see if anything was detected. Loss curves are here: #10, yours look reasonable.

Hi! First of all check if provided pre-trained model works for validation. This will say if validation works. Then check output file with prediction results from your checkpoint. If it is empty, then such error may occur. So, if it is empty, try to visualize the avg_heatmaps to see if anything was detected. Loss curves are here: #10, yours look reasonable.

Thank you for your prompt reply. Thank you for your advice. I'll have a try. I also want to ask you two question. Firstly, the key points I want to set now is 20, and I want to add a waist node. I would like to ask you if I can directly use the COCO data set? However, the COCO data set only annotates 18 key points. If I want to set 20 key points, do I need to annotate the COCO data set again?
Secondly,the key points can be detected if only the third step is carried out,?or the key points and connections can be detected only after the complete training steps have been carried out?
Looking forward to your reply! Thank you!

Yes, you should label somehow waist for the persons, either manually or compute from existing ones. Keypoints can be detected after the third step (you see, there is validation step during training, so after some initial iterations, e.g. 5000, keypoints are detected, try to visualize the heatmaps).

Yes, you should label somehow waist for the persons, either manually or compute from existing ones. Keypoints can be detected after the third step (you see, there is validation step during training, so after some initial iterations, e.g. 5000, keypoints are detected, try to visualize the heatmaps).
Thank you for your prompt reply. I get it!

Was this page helpful?
0 / 5 - 0 ratings

Related issues

hxm1150310617 picture hxm1150310617  ·  4Comments

jinfagang picture jinfagang  ·  18Comments

mathblue picture mathblue  ·  12Comments

augenstern-lwx picture augenstern-lwx  ·  9Comments

tangfayuan picture tangfayuan  ·  7Comments