Skip to content
FusionNet: A deep fully residual convolutional neural network for image segmentation in connectomics
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
merged/medical Initial Commit Apr 29, 2017
result add map results May 2, 2017
Basic_blocks.py Initial Commit Apr 29, 2017
FusionNet.py naming error check Jun 26, 2017
README.md Update README.md May 19, 2017
main.py Update main.py Jun 26, 2017

README.md

FusionNet_Pytorch

FusionNet: A deep fully residual convolutional neural network for image segmentation in connectomics

Requirements

  • Pytorch 0.1.11
  • Python 3.5.2
  • wget

Download code

git clone https://github.com/GunhoChoi/FusionNet_Pytorch
cd FusionNet_Pytorch

Download Map data

wget https://people.eecs.berkeley.edu/~tinghuiz/projects/pix2pix/datasets/maps.tar.gz
tar -xzvf maps.tar.gz

Make model directory

mkdir model

Train Model

python3 main.py  -> medical image without augmentation
python3 main_augmented.py  -> map image with naive augmentation

Out of memory error -> change batch_size / img_size / out_dim

Result

Medical Image

Map Image

Original Image / Label Image / Generated Image

Related Works

You can’t perform that action at this time.