-
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
2D-UNet Image Segmentation of HipMRI slices #157
base: topic-recognition
Are you sure you want to change the base?
2D-UNet Image Segmentation of HipMRI slices #157
Conversation
images to be same size.
…mber of classes in dataset
using nearest neighboor interpolation
This is an initial inspection, no action is required at this point
|
Marking
Marked as per the due date and changes after which aren't necessarily allowed to contribute to grade for fairness. |
Daniel Cottrell |
Hey team I just noticed I only put my student number not my name I'm not sure who Daniel Cottrell is but I'm Jayke Braga. My student number above is correct 46991638. My grades aren't showing up on blackboard either. |
Cant merge because of conflicting changes to main repo files. Please update for merge, doesn't affect grade. |
This repository contains a model which attempts to segment 2D slices of HipMRI data using a UNet.
It contains the following files:
dataset.py - Contains subroutines used for loading and preprocessing data in the nifti format
modules.py - Contains the structure for the UNet model
train.py - The main training script for training the model
predict.py - Used for model inference by loading a set a model weights
graphs - folder containing predicted and true segmentation masks, dice loss during training and images used in the README
Student Number- 46991638