VCG / parallel_unet

2D parallel arms of 3D unet for malis.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

parallel_unet

Implementation of 2D parallel arms of a 3D unet with a common loss layer in caffe. The networks are generated by the mknet.py script and the training is performed with the train.py script. Make sure to update the path to the data files in train.py

About

2D parallel arms of 3D unet for malis.


Languages

Language:C++ 57.0%Language:Makefile 19.8%Language:Jupyter Notebook 18.2%Language:Python 4.9%Language:Cuda 0.1%Language:Shell 0.0%Language:C 0.0%