weixin_39550486
weixin_39550486
2020-12-02 14:09

implement IOU Loss

Based on: https://github.com/Zzh-tju/DIoU-SSD-pytorch/blob/86a370aa2cadea6ba7e5dffb2efc4bacc4c863ea/utils/box/box_utils.py#L47 Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression https://arxiv.org/pdf/1911.08287.pdf Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regression https://giou.stanford.edu/GIoU.pdf UnitBox: An Advanced Object Detection Network https://arxiv.org/pdf/1608.01471.pdf

I made a quick test using the global wheat detection dataset, the AP.5..0.75 reaches 0.35 and stop to increase.

该提问来源于开源项目:rwightman/efficientdet-pytorch

  • 点赞
  • 回答
  • 收藏
  • 复制链接分享

13条回答

为你推荐

换一换