/Volumetric-ConvNet-for-prostate-segmentation

Network architecture for "Volumetric ConvNets with Mixed Residual Connections for Automated Prostate Segmentation from 3D MR Images"

Primary LanguageShell

Volumetric-ConvNet-for-prostate-segmentation

This repository contains the network architecture and training script for network in "Volumetric ConvNets with Mixed Residual Connections for Automated Prostate Segmentation from 3D MR Images".

Caffe library

We slightly modify the original Caffe library to supporting the 3D operations. You can find the Caffe library in https://github.com/yulequan/3D-Caffe.

For installation, please follow the offical instructions of Caffe.

Note

  • We use HDF5DataLayer to read data. You need to generate the hdf5 data from original data type.