我有兴趣用空间变换器网络尝试,我无法在caffe或tensorflow中找到它的任何实现,这是我唯一感兴趣使用的两个库。我对tensorflow有很好的把握,但是想知道是否可以直接用tensorflow提供的现有构建块来实现,而不必像编写自定义c ++模块那样做太复杂的事情
答案 0 :(得分:5)
是的,使用现有API为空间变压器网络设置Tensorflow图表非常简单。
您可以在Tensorflow中找到一个示例实现[1]。
[1] https://github.com/daviddao/spatial-transformer-tensorflow
答案 1 :(得分:1)
这里有一个caffe的实现。 https://github.com/daerduoCarey/SpatialTransformerLayer
答案 2 :(得分:1)
Tensorflow在模型库中实现了空间变换器网络 - https://github.com/tensorflow/models/tree/master/research/transformer