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

Result not showing in the output folder #1

Open
IrvingF7 opened this issue Aug 20, 2020 · 2 comments
Open

Result not showing in the output folder #1

IrvingF7 opened this issue Aug 20, 2020 · 2 comments

Comments

@IrvingF7
Copy link

Hi!
I am trying to run the classifier based on YOLOv3, and after I run it with yolov3.weights the outputs are just the original pics. If I run it with yolov3.pt it would show error msg like this
size mismatch for module_list.105.Conv2d.bias: copying a param with shape torch.Size([255]) from checkpoint, the shape in current model is torch.Size([21]).

I am thinking that the trained model that I am using is not the intended model for gender detection.

BTW I am not using docker as docker build command always fails. Do you think that may be the problem?

Thanks!

@Pari-singh
Copy link
Owner

Hi @IrvingF7, thanks for your interest in my work.

I am not sure where you are taking the weights yolov3.pt from, but I suspect it is a yolo detector most likely trained on COCO classes, which doesn't have man vs woman. It has one class for Person, including both the genders.

Let me know if any further doubts.

@IrvingF7
Copy link
Author

Hi!
Thank you for your reply! I apologize for the late reply. For some reasons Github is not sending me notification.

I am getting yolov3.pt and yolov3.weights from Releases of this repo under the tag name "weights".
After reading your description in Yolov3_gender_detection I was under the impression that there is a trained detector ready to detect men vs women so I looked around and was only able to find that one.

If possible could you provide a trained model that can run on the sample images to detect men and women? Really appreciate 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