当前位置:   article > 正文

yolov5训练的准确率比较低,召回率比较高的问题?_yolov5突然准确率降低

yolov5突然准确率降低

原因:我在使用下面的程序划分数据集时,发现划分之后数据集对应的图片和标签的数量不对应。(自己犯的一个初级错误)。

  1. import os
  2. import shutil
  3. import random
  4. # 训练集、验证集和测试集的比例分配
  5. test_percent = 0.1
  6. valid_percent = 0.09
  7. train_percent = 0.81
  8. # 标注文件的路径
  9. image_path = 'images'
  10. label_path = 'labels'
  11. images_files_list = os.listdir(image_path)
  12. labels_files_list = os.listdir(label_path)
  13. print('images files: {}'.format(images_files_list))
  14. print('labels files: {}'.format(labels_files_list))
  15. total_num = len(images_files_list)
  16. print('total_num: {}'.format(total_num))
  17. test_num = int(total_num * test_percent)
  18. valid_num = int(total_num * valid_percent)
  19. train_num = int(total_num * train_percent)
  20. # 对应文件的索引
  21. test_image_index = random.sample(range(total_num), test_num)
  22. valid_image_index = random.sample(range(total_num), valid_num)
  23. train_image_index = random.sample(range(total_num), train_num)
  24. for i in range(total_num):
  25. print('src image: {}, i={}'.format(images_files_list[i], i))
  26. if i in test_image_index:
  27. # 将图片和标签文件拷贝到对应文件夹下
  28. shutil.copyfile('images/{}'.format(images_files_list[i]), 'test/images/{}'.format(images_files_list[i]))
  29. shutil.copyfile('labels/{}'.format(labels_files_list[i]), 'test/labels/{}'.format(labels_files_list[i]))
  30. elif i in valid_image_index:
  31. shutil.copyfile('images/{}'.format(images_files_list[i]), 'valid/images/{}'.format(images_files_list[i]))
  32. shutil.copyfile('labels/{}'.format(labels_files_list[i]), 'valid/labels/{}'.format(labels_files_list[i]))
  33. else:
  34. shutil.copyfile('images/{}'.format(images_files_list[i]), 'train/images/{}'.format(images_files_list[i]))
  35. shutil.copyfile('labels/{}'.format(labels_files_list[i]), 'train/labels/{}'.format(labels_files_list[i]))

解决办法:我手动划分了数据集,这样确保了数据集的图片和标签是对应关系。图片的比例是训练集:验证集:测试集=0.81:0.09:0.1

迷惑:我第一次用这个代码划分数据集,还是正确的,但是第二次用这个代码划分数据集就是错误的。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/463171
推荐阅读
相关标签
  

闽ICP备14008679号