Retinanet-Pytorch icon indicating copy to clipboard operation
Retinanet-Pytorch copied to clipboard

detect测试,很多框都是重合的,是不是没有nms?

Open yangbisheng2009 opened this issue 4 years ago • 1 comments

yangbisheng2009 avatar Nov 16 '19 07:11 yangbisheng2009

有的, 你可以通过 NMS_THRESHOLD 超参数来 调整 .

nms直接调用的 torchvision 的方法.

具体情况 你可以看 Model / struct / Postprocess. py 里面

yatengLG avatar Nov 16 '19 08:11 yatengLG