-
Notifications
You must be signed in to change notification settings - Fork 284
Siamese network for melanoma classification - 48515739 #292
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
base: topic-recognition
Are you sure you want to change the base?
Siamese network for melanoma classification - 48515739 #292
Conversation
…ase for my solution. Prompts and complete answers from Google's Gemini are included for completeness and transparency
…ng in current state (debugging and running happens on another machine, where repo needs to be cloned to be accessed). The code mostly mixes the Gemini outputs, with additional self-coded parts. Proper data storage and loading still mostly missing
…ining to normal triplet loss (models didn't learn) - more complex embedding network (now a modified ResNet50) - simplified the classifier head - oversample training data but sample the training set for faster training
…etting, results are mildly better
…evious commit, previous commit message applies to this commit as well
…in.py + cleaned up imports in the other files
…ts, along with a few other details
This is an initial inspection, no action is required at this pointRecognition Problem : total : 19.5
Good design: design is fairly strong(1) Overall Great work, thanks Melissa ! |
Marking
Marked as per the due date and changes after which aren't necessarily allowed to contribute to grade for fairness. |
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
This pull request implements a Siamese network for the classification of the ISIC 2020 Kaggle Challenge data set, a dataset of benign and malignant skin lesions.
Th code implements this functionality in python, using the Pytorch library. It introduces four files:
dataset.pyfor data management (splitting augmentation and retrieval)modules.pyfor network architecture and loss functions (ResNet50 with classifier head, triplet loss)predict.pyfor model evaluation (Accuracy, ROC, AUPRC, t-SNE visualisation)train.pyfor training functions and main functionality (main function trains and tests a model according to hyperparameters specified in the file)Please refer to the README file for more details on the functionality as well as a discussion on model results.
Thank you
Melissa Maillot - 48515739