Reputation: 5563
Just curious, are there any Tensorflow implementation of SegNet. Looks like there are multiple Caffe implementation outside. Any information or insight will be appreciated.
Upvotes: 2
Views: 3689
Reputation: 5929
Here is one implementation on github: https://github.com/tkuanlun350/Tensorflow-SegNet
It is a little bit different because the indices unravel has not be achieved in tensorflow, the corresponding questions: In Tensorflow, how to unravel the flattened indices obtained by tf.nn.max_pool_with_argmax? and In Tensorflow, how to assign values in Tensor according to the indices?
Upvotes: 1