tensorflow2.3实践 猫狗识别(详细步骤注释及93%的准确率)

系统
ubuntu 20.04

IDE
pycharm(社区版)
ps:建议专业版,真能避免许多麻烦

代码

import tensorflow as tf
import matplotlib.pyplot as plt
import pathlib
import random

#提取所有文件路径
data_dir = '/...../test'
data_root = pathlib.Path(data_dir)
all_image_path = list(data_root.glob('*/*'))
all_image_path = [str(path) for path in all_image_path]
#乱序
random.shuffle(all_image_path)
#记录总数,以备后用
image_count = len(all_image_path)

#编码标签
#Cat 0  Dog 1
label_names = sorted(item.name for item in data_root.glob('*/'))
label_to_index = dict((name,index) for index,name in enumerate(label_names))

#给图片贴标签
#pathlib.Path('..../2.jpg').parent.