试图解决的问题的背景:
我正在使用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文件。 让我知道是否需要更多详细信息。