zjykzj / guestbook

Welcome to give me advice
0 stars 0 forks source link

[目标检测][PyTorch]边界框操作 | 大海 #38

Open zjykzj opened 3 years ago

zjykzj commented 3 years ago

https://blog.zhujian.life/posts/c306db14.html

阅读源码时发现torchvision提供了许多边界框操作,包括 NMS 移除小边界框 修剪边界框坐标,保证边界框坐标位于图像内 计算边界框面积 IoU计算 在线文档:torchvision.ops 源码地址:torchvision/ops/boxes.py 当前torchvision版本:0.6.0a0+82fd1c8