CODEWITHSUNDEEP
caffematcaffe
user570593
user570593

Reputation: 3520

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?

Upvotes: 0

Views: 877

Answers (1)

malreddysid
malreddysid

Reputation: 1342

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.

Upvotes: 1

Related Questions