深度学习-Tensorflow2.2-多分类{8}-多输出模型实例-20

``

import tensorflow as tf
from tensorflow import keras
import matplotlib.pyplot as plt
%matplotlib inline
import numpy as np
import pathlib
import os
import random
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
import IPython.display as display

gpu_ok = tf.test.is_gpu_available()
print("tf version:", tf.__version__)
print("use GPU", gpu_ok) # 判断是否使用gpu进行训练

data_dir = "F:/py/ziliao/数据集/multi-output-classification/dataset" # 定义路径

data_root = pathlib.Path(data_dir)
data_root

for item in data_root.iterdir():
    print(item) # 查看所有目录

all_image_paths = list(data_root.glob("*/*")) # 使用glob方法取出所有图片

image_count = len(all_image_paths)
image_count # 查看图片张数

all_image_paths = [str(path) for path in all_image_paths]
random.shuffle(all_image_paths)  # 对所有图片路径进行乱序

# 提取目录名称
label_names = sorted(item.name for item in data_root.glob("*/") if item.is_dir(