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

Trained Weights #3

Open
alevangel opened this issue Jan 20, 2024 · 3 comments
Open

Trained Weights #3

alevangel opened this issue Jan 20, 2024 · 3 comments

Comments

@alevangel
Copy link

Where is possible to download the pretrained weights?

@mkang315
Copy link
Owner

yolov5l-seg.pt that contains pretrained weights has already been uploaded.

@alevangel
Copy link
Author

alevangel commented Jan 31, 2024

@mkang315 taking image 9 of the BCC validation dataset as an example, I can't get any predictions for the trsh=0.25. Why this may happen?

This is the code in order to repeat :
python segment/predict.py --weights yolov5l-seg.pt --source datasets/BCC/images/val/9.bmp

If I reduce the threshold to 0.0001 instead, I get this result:
Cattura

@mkang315
Copy link
Owner

mkang315 commented Feb 1, 2024

Sorry, I misunderstood what you mean. The file yolov5l-seg.pt contain the initial weights that I described in the paper and isn't the pretrained weights of ASF-YOLO. I know you want best.pt of ASF-YOLO that I haven't it for now, but you can train the model on the train set and then test examples.

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