IOU计算式具体在哪？

bubbliiiing / faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库，可以利用voc数据集格式的数据进行训练。

MIT License

1.56k stars 352 forks source link

Open liusas12 opened 1 year ago

liusas12 commented 1 year ago

导师好请问在您的faster rcnn代码中，怎样修改iou计算方式，我看到def nms函数中只有个返回值torch.ops.torchvison.nms(),请问iou具体的计算式在哪。

bubbliiiing commented 1 year ago

如果要修改nms的代码，那么要重写nms

liusas12 commented 1 year ago

b导为啥def nms函数中间全是注释没有代码在哪重写

---原始邮件--- 发件人: @.> 发送时间: 2023年6月1日(周四) 晚上11:39 收件人: @.>; 抄送: @.**@.>; 主题: Re: [bubbliiiing/faster-rcnn-pytorch] IOU计算式具体在哪？ (Issue #180)

如果要修改nms的代码，那么要重写nms

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

bubbliiiing commented 1 year ago

因为原来的nms是自己实现的，慢，用了官方的