search

SeanCho1996 / Non_maximum_suppression_for_3D

implementation of NMS algorithm for 3D image usage
2 stars 0 forks source link

yolo_v3 3D detect #1

Open Luoxiaoyu828 opened 3 years ago

Luoxiaoyu828 commented 3 years ago

你好，请问将yolo_v3由二维检测升到三维检测后 你的这个NMS算法可以直接替换掉二维中的NMS算法吗？以及是如何使用的，我现在还没有具体使用。

SeanCho1996 commented 3 years ago

在模型外部可以直接使用，但是在深度学习框架中可能需要移植到框架内，例如torch里需要写成init+forward这种形式

Luoxiaoyu828 commented 3 years ago

我是要在tensorflow上用 感觉直接使用会有一些问题 如果要用的话需要需要做一些什么处理 有哪些需要注意的 请教一下

SeanCho1996 commented 3 years ago

正常情况下nms应该是在预测结束之后清理预选框时的操作，理论上来说不需要放在框架内部，也不需要太复杂的操作，可以先试一下效果，有bug再解决bug。这个代码也是我上学时候写的，可能还有点稚嫩，哈哈哈

Luoxiaoyu828 commented 3 years ago

那也行 我这段时间先实现一下 如果有问题再请教你 一起讨论