ISIC2018 Loading and Training Configuration for YOLOv7 Lesion Detection Model - #191

Open

pri-gression wants to merge 47 commits into

shakes76:mainfrom

pri-gression:main

Commits on Oct 29, 2024

Initial commit for COMP3710
pri-gression
committed
04e3ec4
Initial commit test for COMP3710
pri-gression
committed
040922f

Commits on Oct 30, 2024

Cloned YOLOv7 repo
pri-gression
committed
182ab96
Removed hello.py and updated repository files
pri-gression
committed
44acbf2
Add yolov7 as a regular folder
pri-gression
committed
d84aaaa
Added .gitignore to ignore dataset and system files
pri-gression
committed
b83a0d6
Removed .DS_Store from tracking
pri-gression
committed
9ab3b6d
Loaded and preprocessed ISIC dataset for training
pri-gression
committed
b90aee3
Loaded and preprocessed ISIC dataset for training
pri-gression
committed
0066fc6

Commits on Oct 31, 2024

Source code for model components
pri-gression
committed
7b34a0c
Trained, Validated and tested the model
pri-gression
committed
28b8133
Used to predict and visualised data via predict.py
pri-gression
committed
3b0cd4b
Updated the path settings to point to the linked ISIC2018 directory
pri-gression
committed
50e617a
Addresses GPU device error in training.py
pri-gression
committed
b13ab90
Changed the initialization of self.device in the LesionDetectionModel class
pri-gression
committed
b7fc327
Fixed Model Instantitation
pri-gression
committed
8afed8a
Added a checkpoint to only extract model weights from yolov7.pt file
pri-gression
committed
5d60937
Adjusted code to load full model correctly
pri-gression
committed
3ce2779
Successfully loaded the model with yolov7
pri-gression
committed
0862a1d
Handled segmentation mask files in .png instead of .json
pri-gression
committed
09a0158
Applied resizing transform to masks and images
pri-gression
committed
a13ae1d
Converted Bounded box coordinates to YOLO format
pri-gression
committed
e9b3e65

Commits on Nov 1, 2024

Adjusted the target tensor so it matches the outputs
pri-gression
committed
a23b863
README.md test
pri-gression
committed
02c515d
README.md updated
pri-gression
committed
fd7cb5c
YOLOv7 Architecture Explained
pri-gression
committed
dfce478
Image Fixed on README.md
pri-gression
committed
6eb26c5
Uploaded Image for README.md
pri-gression
committed
54e1222
Image Uploaded Test
pri-gression
committed
da2c71b
Image Upload Test
pri-gression
committed
98e2be4
README.md update
pri-gression
committed
9eb4e05
Moved self.img_size = img_size before calling self.default_transforms() (AttributeError)
pri-gression
committed
695f015
Adding transfer learning by freezing layers
pri-gression
committed
0cd9d4b
Training with Learning Rate Scheduling, Early Stopping
pri-gression
authored
538106b
IoU Calculation added
pri-gression
authored
d4b98a2
README.md updated
pri-gression
committed
0be5a37
README.md changes
pri-gression
committed
6e1f3ac
README.md final
pri-gression
committed
626b58b
README.md references link fix
pri-gression
committed
a9d25ac

Commits on Nov 2, 2024

Commits on Nov 15, 2024