- 博客(2)
- 收藏
- 关注
RSS订阅原创 用DETR进行目标检测的predicat文件
文章目录前言二、使用步骤1.更改官方detr中308行的类别数2.根目录加入检测文件总结前言由于最近研究DETR目标检测，但是发现官方给的代码缺少了predict文件，于是自己写了一个，但是写代码的过程中也想到了一些问题，比如DETR没有采用NMS，该如何避免预测框重叠。总之按照论文方法简单实现了一下DETR的predict。一、DETR的predict实现过程由于只是测试用，仅仅简单实现了一下，有其他需要的可以自己更改，过程比较简单，就是加载训练权重，图像预处理送入网络进行前传，对得到的pr
2022-04-17 19:54:20
4047
22
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人