关于cats_vs_dogs猫狗数据集训练错误

今天拿tf搭模型跑猫狗数据集的时候差点被逼疯，fit()的时候每次都中断出错，后来发现是数据集的问题。。。。。

num_skipped = 0
for folder_name in ("Cat", "Dog"):
    folder_path = os.path.join("PetImages", folder_name)
    for fname in os.listdir(folder_path):
        fpath = os.path.join(folder_path, fname)
        try:
            fobj = open(fpath, "rb")
            is_jfif = b"JFIF" in fobj.peek(10)
        finally:
            fobj.close()

        if not is_jfif:
            num_skipped += 1
            # Delete corrupted image
            os.remove(fpath)

print(f"Deleted {num_skipped} images.")

搬运一下keras上面的解决方案。。。。。谁能想到。。。