迁移学习 colab 完整示例:fruits-360 数据集

这里当前目录下已经有fruits-360这个数据集. 关于调用数据集的方法可以查看我另一篇文章.

准备

import tensorflow as tf
import tensorflow.keras as keras
from tensorflow.keras.preprocessing.image import load_img, img_to_array, array_to_img, ImageDataGenerator

创建 Generator

创建 ImageDataGenerator. 由于这个数据集足够大, 所以不需要进行 image augmentation.

train_datagen = ImageDataGenerator(rescale=1./255)
test_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(
        "fruits-360/Training",
        target_size=(100, 100