检测训练文件夹中无效的图像

在训练过程序中，有的文件夹中图片是无效的，会导致训练崩溃，以下代码实现了检测并删除

from tensorflow.python.framework import graph_util
from tensorflow.python.framework import tensor_shape
from tensorflow.python.platform import gfile
from tensorflow.python.util import compat
import keras
import os
import time

def get_file_path(root_path,file_list,dir_list):
    #获取该目录下所有的文件名称和目录名称
    dir_or_files = os.listdir(root_path)
    for dir_file in dir_or_files:
        #获取目录或者文件的路径
        dir_file_path = os.path.join(root_path,dir_file)
        #判断该路径为文件还是路径
        if os.path.isdir(dir_file_path):
            dir_list.append(dir_file_path)
            #递归获取所有文件和目录的路径
            get_file_path(dir_file_path,file_list,dir_list)
        else:
            file_list.append(dir_file_path)



root_path = r"D:\\working\\Python\\readHeadFileTraining\\Testflower\\tf_files\\flower_photos"
#用来存放所有的文件路径
file_list = []
#用来存放所有的目录路径
dir_list = []
get_file_path(root_path,file_list,dir_list)
#print(file_list)
print(dir_list)
for idx, image_path in enumerate(file_list):
    try:
        image_data = gfile.FastGFile(image_path, 'rb').read()
        #print(idx)
    except:
        print(image_path)
        os.remove(image_path)


print("ok finished")