纯张量流中NMS的定制实现(使用Fasterrcnn进行点检测)

时间:2019-02-11 22:18:40

标签: tensorflow deep-learning

试图解决的问题的背景:

我正在使用FasterRcnn检测关键点。我被困在tensorflow中自定义NMS的实现上。我不能将tensorflow的NMS用于边界框,而是想将其实现为基于欧氏距离作为参数的关键点检测。

我看到github仓库是他们实现了自定义NMS的视图,但是在caffe中不是tensorflow: https://github.com/msracver/Deformable-ConvNets

https://github.com/kevinjliang/tf-Faster-RCNN

所以我的问题:是否可以为纯tensorflow中的rpn层到rcnn层的提案实施自定义NMS?

我想在tensorflow中编写整个NMS操作,我被困的地方是在不使用sess.run()的情况下访问NMS逻辑的单个索引。

PS:我不知道CUDA编程,不能编辑上述存储库中提到的.cu文件。 让我知道是否需要更多详细信息。

0 个答案:

没有答案