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

No Detection of images even after training 1200 images with 100 epochs. #2075

Open
theunbeatenseven opened this issue Aug 20, 2024 · 0 comments

Comments

@theunbeatenseven
Copy link

My goal was to train model to predict eye disease . I collected dataset of 1500 images and did 60 20 20 split. I trained my model with the following command "python train.py --workers 1 --device 0 --batch-size 4 --epochs 100 --img 640 640 --data data/custom_data.yaml --hyp data/hyp.scratch.custom.yaml --cfg cfg/training/yolov7-custom.yaml --name yolov7-custom --weights yolov7.pt".
I modified yolov7-custom.yaml by creating 2 classed namely healthy and glaucoma(diseased).
After training I tried detecting one of the images in training and test dataset but encountered no detection.
These are some of the files in my runs that might require for finding the bug.
confusion_matrix
F1_curve
P_curve
PR_curve
R_curve
results
Why there is no detection and how can I make my model detect and produce some output. Also, How can I continue from 100 epoch trained model? Thank you for your replies in advanced.

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

1 participant