python opencv遍历图像数据集是否存在错误

python3.9的环境，opencv3.4:

平时在准备图像数据集是，有可能其中有个别图像错误引起在深度学习训练到一半时报错，所有先检查一下数据集中的图像是否有错误图像：

1. import os
   
2. import cv2
   
3. import glob
   
4. import pathlib
   
5. import random
   
6. import numpy as np
   
7. 
   
8. from os import listdir, getcwd
   
9. 
   
10. from os.path import join
    
11. 
    
12. # 检查图片数据是否正确
    
13. 
    
14. # 图片路径
    
15. data_path = r'E:\\datasets\\CCPD2019-yolo'
    
16. 
    
17. # 写入到文件
    
18. trainfile = r'E:\\datasets\\err_list.txt'
    
19. 
    
20. if __name__ == '__main__':
    
21. 
    
22.     file_list = os.listdir(data_path)
    
23. 
    
24.     train_file = open(trainfile, 'a',encoding='utf-8-sig') #带BOM的UTF-8格式
    
25.     for file_obj in glob.glob(data_path + '/*.jpg', recursive=True):
    
26.     #for file_obj in file_list:
    
27.         file_path = os.path.join(data_path, file_obj)
    
28.         if os.path.exists(file_path):
    
29.             a=1
    
30.             #print(file_path)
    
31.         else:
    
32.             print('不存在', file_path)
    
33.             continue
    
34.         # img=cv2.imread(img_path)
    
35.         img = cv2.imdecode(np.fromfile(file_path, dtype=np.uint8), -1)  # 解决读中文名称图片的问题
    
36.         if img is None:
    
37.             print('错误:%s', file_path)
    
38.             d = pathlib.Path(file_path)
    
39.             train_file.write(file_path + '\t' + 'carplate' + '\n')
    
40.             continue
    
41. 
    
42.     train_file.close()

复制代码