Retinanet-Pytorch
Retinanet-Pytorch copied to clipboard
detect测试,很多框都是重合的,是不是没有nms?
有的, 你可以通过 NMS_THRESHOLD 超参数来 调整 .
nms直接调用的 torchvision 的方法.
具体情况 你可以看 Model / struct / Postprocess. py 里面
有的, 你可以通过 NMS_THRESHOLD 超参数来 调整 .
nms直接调用的 torchvision 的方法.
具体情况 你可以看 Model / struct / Postprocess. py 里面