YOLO训练得到权重后无法检测detect目标

  1. 通过自己制造数据集,跑完train.py文件后,得到自己的权重文件

  1. 将权重文件带入detect.py文件中,发现可以运行,但是无法识别图片和视频中的目标

3.opencv-python版本太高了,看了一眼,果然版本都到4.6了,猜想opencv-python版本问题,结果——还是不行

https://blog.csdn.net/adai5210/article/details/128271384?utm_medium=distribute.pc_relevant.none-task-blog-2~default~baidujs_utm_term~default-2-128271384-blog-125722979.pc_relevant_vip_default&spm=1001.2101.3001.4242.2&utm_relevant_index=5

4.在detect.py里面增加:cudnn.benchmark = True——没用。

https://blog.csdn.net/hh571050143/article/details/119984840?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-1-119984840-blog-128271384.pc_relevant_vip_default&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-1-119984840-blog-128271384.pc_relevant_vip_default&utm_relevant_index=2

5.训练次数不够,增加到300,parser.add_argument('--epochs', type=int, default=300)

https://blog.csdn.net/todo_cct/article/details/125722979?ops_request_misc=&request_id=&biz_id=102&utm_term=yolov5%E7%9A%84%E7%9B%AE%E6%A0%87%E6%97%A0%E6%B3%95%E8%AF%86%E5%88%AB&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-9-125722979.142^v73^control,201^v4^add_ask,239^v2^insert_chatgpt&spm=1018.2226.3001.4187

结果——可以

所以,增加训练次数确实会有助于提高识别率

猜你喜欢

转载自blog.csdn.net/qq_56508207/article/details/129543912
今日推荐