Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

train No.3 #154

Closed
RCpengnan opened this issue Mar 21, 2021 · 6 comments
Closed

train No.3 #154

RCpengnan opened this issue Mar 21, 2021 · 6 comments

Comments

@RCpengnan
Copy link

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

@RCpengnan
Copy link
Author

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

@RCpengnan
Copy link
Author

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

@Daniil-Osokin
Copy link
Owner

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.

@RCpengnan
Copy link
Author

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!

@Daniil-Osokin
Copy link
Owner

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).

@RCpengnan
Copy link
Author

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!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants