FusionNet: A deep fully residual convolutional neural network for image segmentation in connectomics
- Pytorch 0.1.11
- Python 3.5.2
- wget
git clone https://github.com/GunhoChoi/FusionNet_Pytorch
cd FusionNet_Pytorch
wget https://people.eecs.berkeley.edu/~tinghuiz/projects/pix2pix/datasets/maps.tar.gz
tar -xzvf maps.tar.gz
mkdir 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
Original Image / Label Image / Generated Image