Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
tianma 4f2a513b53 | 1 year ago | |
---|---|---|
.. | ||
csrc/rbox_iou | 1 year ago | |
unittest | 1 year ago | |
README.md | 1 year ago | |
setup.py | 1 year ago |
旋转框IOU计算OP是参考自定义外部算子 。
python setup.py install
编译完成后即可使用,以下为rbox_iou
的使用示例
# 引入自定义op
from ext_op import rbox_iou
paddle.set_device('gpu:0')
paddle.disable_static()
rbox1 = np.random.rand(13000, 5)
rbox2 = np.random.rand(7, 5)
pd_rbox1 = paddle.to_tensor(rbox1)
pd_rbox2 = paddle.to_tensor(rbox2)
iou = rbox_iou(pd_rbox1, pd_rbox2)
print('iou', iou)
可以通过执行单元测试来确认自定义算子功能的正确性,执行单元测试的示例如下所示:
python unittest/test_matched_rbox_iou.py
基于百度飞桨PaddleDetection套件中的基础模型YOLOv3进行开发的垃圾目标检测模型,能够精准地识别日常生活中常见的11种垃圾。
Text Python Jupyter Notebook Roff C++ other
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》