/parallel_unet

2D parallel arms of 3D unet for malis.

Primary LanguageC++

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