-
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
Segmenting MRI images using 2D Unet #142
base: topic-recognition
Are you sure you want to change the base?
Segmenting MRI images using 2D Unet #142
Conversation
…t was just the unsegmented data, added import calls to train.py to access each dataset
The encoder computes conv's and max pooling and saves the skip connections The bottleneck computes two convolutions with 1024 features The decoder apply upconv's, adds the skip connections and does convolutions The unet function groups each of these together and makes a model
Modified the input shape in modules.py, the model sucessfully runs epochs. Going to add the valdation and testing to modules.py as well as dice similarity
… the training, validation and testing of the model to the train.py. Added imports and plots to the predict.py so that the results of the model could be visualised. Going to add prints of the true mask vs the predicted mask to show in the readme Havent successfully tested the model yet
…RANGPUR. Going to impliment a custom dice function to calculate the dice score. After the results from Rangpur are found I will tweak the model accordingly
true segmentation and predicted segmentations Current predictions are very poor, however only 1/10 of the data is being used to train as my local machine is running the code. going to modify the train.py to include working dice
Added dice to train.py referenced it aswell Increased the learning rate to 0.0001 cleaned up unuesd imports
Removed mid function comments from modules to make it less busy Tried multiple different loss functions however I cannot get better dice than around 0.4 If dice loss is used then the preducted segmentation is black
I added data sorting as per someones ed post and it made the model significantly better, also changedd the folder layout to suit the spec
Changed to dice and binary cross entropy, getting 65 dice off 3 epochs of all the data
Accidently removed files from github thinking they were duplicates.
going to run with the full dataset now to get the performance and then will work on readme
This is an initial inspection, no action is required at this point
Note |
Nicholas Marchant |
Marking
Marked as per the due date and changes after which aren't necessarily allowed to contribute to grade for fairness. |
Nicholas Marchant |
Cant merge because of conflicting changes to main repo files. Please update for merge, doesn't affect grade. |
Good morning,
Sorry I’m a little confused as to which files need to be updated and which are conflicting.
Kind regards,
Nicholas Marchant
…________________________________
From: Shakes ***@***.***>
Sent: Tuesday, November 19, 2024 2:46:23 PM
To: shakes76/PatternAnalysis-2024 ***@***.***>
Cc: NicMarchant ***@***.***>; Author ***@***.***>
Subject: Re: [shakes76/PatternAnalysis-2024] Segmenting MRI images using 2D Unet (PR #142)
Cant merge because of conflicting changes to main repo files. Please update for merge, doesn't affect grade.
—
Reply to this email directly, view it on GitHub<#142 (comment)>, or unsubscribe<https://github.com/notifications/unsubscribe-auth/BHWFOR6JDUFD4MQ2FWJ3SST2BLGC7AVCNFSM6AAAAABQUOYEYWVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDIOBUG42TQNJYG4>.
You are receiving this because you authored the thread.Message ID: ***@***.***>
|
This pull request implements a 2D Unet to create binary class segmentation of hip MRI images.