深度学习-Tensorflow2.2-卷积神经网络{3}-卫星图像识别卷积综合实例（二分类）-13

import tensorflow as tf
import matplotlib.pyplot as plt
%matplotlib inline
import numpy as np
import pathlib

数据读取及预处理

data_dir = "./2_class"# 文件路径

data_root = pathlib.Path(data_dir)# 构建路径对象

for item in data_root.iterdir(): # 对目录进行迭代查看文件路径及对象
    print(item)

all_image_path = list(data_root.glob("*/*"))#使用glob方法及正则表达式提取目录里面所有文件
len(all_image_path) # 1400个数据

all_image_path[:3]# 通过切片查看前3个文件

all_image_path = [str(path) for path in all_image_path]# 使用str把路径变成一个实际的路径
all_image_path[10:12]

import random
random.shuffle(all_image_path)# 把内容乱序
all_image_path[10:12]

image_count = len(all_image_path)
image_count # 记录图片的张数

label_names = sorted (item.name for item in data_root.glob("*/")) # 提取分类名字
label_names

# 编码airplane对应0, lake对应1
label_to_index = dict((name,index) for index,name in enumerate(label_names))
label_to_index

all_image_path[:3]

pathlib.Path("2_class\\lake\\lake_405.jpg").parent.name

all_image_label = [lab