-
Notifications
You must be signed in to change notification settings - Fork 179
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
ISIC2018 Loading and Training Configuration for YOLOv7 Lesion Detection Model #191
base: main
Are you sure you want to change the base?
Conversation
This is an initial inspection Difficulty : Normal
Use markdown format in writing PR Feedback:
|
Hi Shaivika, Thank you for your feedback. By time limitations I meant that during those two weeks I was down with CoVid and had exceed ed the 7 day extension period. I will incorporate the feedback. I just wanna ask whether or not am I allowed to make changes to my code, if allowed train it again and push the new changes with a trained model. Also the extra files added are yolov7's directory that I cloned from github and added in the same directory as my code. Thats why there are uploaded. Ill sort it out as well. Thank you so much! |
Observational Feedback Pull Request: File Organizing: Commit Log: Documentation: |
Hi @pri-gression |
Marking
Marked as per the due date and changes after which aren't necessarily allowed to contribute to grade for fairness. |
Feedback marks possible +2 if the requested changes are made (see above). |
Fixed the file structure and made a new pull request. Thank you for your feedback. |
Description:
This pull request adds a YOLOv7-based deep learning model for lesion detection in dermoscopic images from the ISIC 2017/2018 dataset, with the following components:
Model Architecture (modules.py)
Implements the LesionDetectionModel class, which loads a YOLOv7 model pre-trained on COCO weights via PyTorch Hub.
Optimizes for efficient and accurate lesion detection on dermoscopic images.
Initializes the model on the designated device (CPU or GPU), with frozen backbone layers to focus training on lesion-specific learning.
Data Processing (dataset.py)
ISICDataset class handles loading and preprocessing of dermoscopic images and annotations.
Data augmentation, including random horizontal and vertical flips and color jitter, is applied to improve robustness.
Each image is resized to 640x640, normalized with ImageNet stats, and converted to a tensor for model compatibility.
Training and Validation Script (training.py)
Initializes the model, loads training and validation data, and executes the training loop.
Uses binary cross-entropy with logits (BCEWithLogitsLoss) as the loss function and the Adam optimizer with a learning rate scheduler.
The model is trained for 10 epochs with validation checks each epoch, though training has not successfully completed due to time limitations.
Prediction Script (predict.py)
Loads the trained YOLOv7 model for inference on new images.
Processes each test image with resizing, normalization, and model inference.
Visualizes predictions by overlaying bounding boxes for detected lesions on the original images.
Documentation (README.md)
Provides instructions for setting up the environment, running training and prediction scripts, and adjusting configurations as needed.
Specifies model dependencies and their versions to ensure compatibility.
Thank you
Pritika Guglani
47604364