Search code examples
caffematcaffe

Fully convolutional net in Caffe

Is it possible for me to directly give an image and its segmentation as the input for training a caffe? Does the original implementation support this? If yes where can I find an appropreate prototxt file?

Solution

  • Yes. It is possible.

    Have a look at Fully Convolutional Networks for Semantic Segmentation and SegNet. They are both fully convolutional networks and are trained for semantic segmentation. The prototxt and caffemodel files are available on GitHub.

    You can run FCN with the original implementation, but SegNet uses some layers which are not part of the original implementation. They have an edited version of caffe on Github, so you can use that.