纯tensorflow中NMS的自定义实现(使用Fasterrcnn进行点检测)

纯tensorflow中NMS的自定义实现(使用Fasterrcnn进行点检测),tensorflow,deep-learning,Tensorflow,Deep Learning,试图解决的问题的背景: 我正在使用FasterRcnn检测关键点。我一直在tensorflow中实现定制NMS。我不能使用tensorflow的NMS,因为它是用于边界框的,我想将它用于基于欧几里德距离作为参数的关键点检测 我看到查看github回购协议是否实施了自定义NMS,但在caffe中不是tensorflow: 所以我的问题是:是否可以在纯tensorflow中为从rpn层到rcnn层的提案实施自定义NMS 我想在tensorflow中编写整个NMS操作,我的重点是在没有sess.r

试图解决的问题的背景:

我正在使用FasterRcnn检测关键点。我一直在tensorflow中实现定制NMS。我不能使用tensorflow的NMS,因为它是用于边界框的,我想将它用于基于欧几里德距离作为参数的关键点检测

我看到查看github回购协议是否实施了自定义NMS,但在caffe中不是tensorflow:

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

我想在tensorflow中编写整个NMS操作,我的重点是在没有sess.run()的情况下访问NMS逻辑的单个索引

PS:我不知道CUDA编程,编辑上面提到的.cu文件。 如果需要更多细节,请告诉我