-
Notifications
You must be signed in to change notification settings - Fork 284
Attention U-Net for OASIS Brain Tissue Segmentation #286
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?
Conversation
Create OASISDataset class to load dataset from files
- Add ConvBlock class to perform double convolution at encoder and decoder stages. - Add AttentionBlock class to filter intermediate-scale features before merging skip connections.
Added a figure illustrating the Attention U-Net architecture and its source information.
Added DiceLoss and CombinedLoss classes for segmentation tasks.
Added dice coefficient calculation for evaluation and updated training epoch function.
Added main function to initialize datasets, loaders, and model.
Added a function to auto-detect label mapping, number of classes, and image size from segmentation masks. Updated NUM_CLASSES to be determined dynamically.
|
<This is an initial inspection, no action is required at this point.> File Organizing: Problem Solving:
Model and functions:
Code design: Functional works. Code comment and docstring:
Difficulty: Easy. Additional Comments:
|
|
Hi there, Please provide your student number or full name; otherwise, we won’t be able to allocate your marks. |
Marking
Marked as per the due date and changes after which aren't necessarily allowed to contribute to grade for fairness. |
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
Check the bottom of the README |
An Attention U-Net model that performs semantic segmentations of brain MRI scans from the OASIS dataset.