mumumu99/unet-segmentation-lumbar

u-net family implemented by pytorch 1.2 for medical image segmentation

LumbarSegmentation

Our motivation is to optimize current models on accuracy and robustness regarding lumbar spine images.

Descriptions

This project contains Pytorch implementation of various basic models.

• UNet proposed in the paper U-Net: Convolutional Networks for Biomedical Image Segmentation

Prerequisties

• Python 3.7
• Pytorch 1.2
• torchsummary
• tb-nightly

Usage

$ python train.py

Results

• Generated samples with combined losses(100th epochs)

• The loss of training/validing dataset