是否可以直接在tensorflow中实现空间变换器网络?

时间:2016-03-02 20:28:50

标签: tensorflow

我有兴趣用空间变换器网络尝试,我无法在caffe或tensorflow中找到它的任何实现,这是我唯一感兴趣使用的两个库。我对tensorflow有很好的把握,但是想知道是否可以直接用tensorflow提供的现有构建块来实现,而不必像编写自定义c ++模块那样做太复杂的事情

3 个答案:

答案 0 :(得分:5)

是的,使用现有API为空间变压器网络设置Tensorflow图表非常简单。

您可以在Tensorflow中找到一个示例实现[1]。

[1] https://github.com/daviddao/spatial-transformer-tensorflow

答案 1 :(得分:1)

答案 2 :(得分:1)

Tensorflow在模型库中实现了空间变换器网络 - https://github.com/tensorflow/models/tree/master/research/transformer