-
通过自己制造数据集,跑完train.py文件后,得到自己的权重文件
![](https://img-blog.csdnimg.cn/img_convert/5c1f4f0b2c394c62819e2179aa2dd7f4.png)
-
将权重文件带入detect.py文件中,发现可以运行,但是无法识别图片和视频中的目标
![](https://img-blog.csdnimg.cn/img_convert/93ab6299fac2a972c2384a45454e631e.png)
3.opencv-python版本太高了,看了一眼,果然版本都到4.6了,猜想opencv-python版本问题,结果——还是不行
4.在detect.py里面增加:cudnn.benchmark = True——没用。
5.训练次数不够,增加到300,
parser.add_argument(‘–epochs’, type=int, default=300)
![](https://img-blog.csdnimg.cn/img_convert/35aa240ccbdf06641dd503f14f453d0c.png)
结果——可以
![](https://img-blog.csdnimg.cn/img_convert/cb5ed0179a5a4c6ccc02d55527661909.png)
![](https://img-blog.csdnimg.cn/img_convert/46586fca18b81982988a9acfb53b4442.png)
所以,增加训练次数确实会有助于提高识别率
版权声明:本文为qq_56508207原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。