# FusionNet-Pytorch

**Repository Path**: cui_jingjing/FusionNet-Pytorch

## Basic Information

- **Project Name**: FusionNet-Pytorch
- **Description**: FusionNet: A deep fully residual convolutional neural network for image segmentation in connectomics
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2019-11-07
- **Last Updated**: 2020-12-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# FusionNet_Pytorch

[FusionNet: A deep fully residual convolutional neural network for image segmentation in connectomics](https://arxiv.org/abs/1612.05360)

## 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
<img src="./result/original_image_185_0.png" width="20%"><img src="./result/label_image_185_0.png" width="20%"><img src="./result/gen_image_185_0.png" width="20%">

### Map Image 

<img src="./result/satel_image_31_41.png" width="100%">

<img src="./result/map_image_31_41.png" width="100%">

<img src="./result/gen_image_31_41.png" width="100%">

Original Image / Label Image / Generated Image

## Related Works

- [FusionNet in Tensorflow by Hyungjoo Andrew Cho](https://github.com/NySunShine/fusion-net)