查看keras自动给文件夹标号

简介: 1 from tensorflow.contrib.keras.api.keras.preprocessing.image import ImageDataGenerator,img_to_array 2 from tensorflow.
 1 from tensorflow.contrib.keras.api.keras.preprocessing.image import ImageDataGenerator,img_to_array
 2 from tensorflow.contrib.keras.api.keras.models import Sequential
 3 from tensorflow.contrib.keras.api.keras.layers import Dense, Dropout, Activation, Flatten
 4 from tensorflow.contrib.keras.api.keras.layers import Conv2D, MaxPooling2D
 5 IMAGE_SIZE = 224
 6 img_rows= 224
 7 img_cols = 224
 8 # 训练图片大小
 9 epochs = 50#原来是50
10 # 遍历次数
11 batch_size = 32
12 # 批量大小
13 nb_train_samples = 256*2
14 # 训练样本总数
15 nb_validation_samples = 64*2
16 # 测试样本总数
17 train_data_dir = 'D:\\pycode\\learn\\data\\train_data\\'
18 validation_data_dir = 'D:\\pycode\\learn\\data\\test_data\\'
19 # 样本图片所在路径
20 FILE_PATH = 'age.h5'
21 
22 train_datagen = ImageDataGenerator(
23     rescale=1. / 255,
24     horizontal_flip=True)
25 
26 test_datagen = ImageDataGenerator(rescale=1. / 255)
27 
28 train_generator = train_datagen.flow_from_directory(
29     train_data_dir,
30     target_size=(img_rows, img_cols),
31     batch_size=batch_size,
32     class_mode='categorical')
33 
34 validation_generator = test_datagen.flow_from_directory(
35     validation_data_dir,
36     target_size=(img_rows, img_cols),
37     batch_size=batch_size,
38     class_mode='categorical')
39 
40 # self.train = train_generator
41 # self.valid = validation_generator
42 print(validation_generator.class_indices)

目录
相关文章
|
存储 固态存储
阿里巴巴率先应用基于3D XPoint™介质的新型存储技术-Intel® Optane™ SSD
近日,阿里巴巴延续同英特尔在存储领域的深度合作,率先应用基于3D XPoint™介质的新型存储技术-Intel® Optane™ SSD。
1035 0
|
10月前
|
Ubuntu Shell 开发者
Docker容器管理
Docker容器管理
115 0
|
10月前
|
监控 安全 Java
JVM工作原理与实战(十):类加载器-Java类加载器
JVM作为Java程序的运行环境,其负责解释和执行字节码,管理内存,确保安全,支持多线程和提供性能监控工具,以及确保程序的跨平台运行。本文主要介绍了扩展类加载器、通过扩展类加载器去加载用户jar包、应用程序类加载器等内容。
139 4
|
10月前
|
存储 运维 监控
基于业务的告警管理最佳实践
本文主要介绍了SLS告警管理中心的业务概念和功能。
351 0
基于业务的告警管理最佳实践
|
10月前
|
C++
【SPSS】两配对样本T检验分析详细操作教程(附案例实战)
【SPSS】两配对样本T检验分析详细操作教程(附案例实战)
908 0
【SPSS】两配对样本T检验分析详细操作教程(附案例实战)
|
10月前
|
数据可视化 安全 搜索推荐
低代码开发:突破还是威胁?探讨低代码的发展与挑战
低代码开发:突破还是威胁?探讨低代码的发展与挑战
153 1
|
算法 C++ iOS开发
解决办法一:GetProcAddress函数返回值总为NULL
解决办法一:GetProcAddress函数返回值总为NULL
282 0
|
算法 搜索推荐
58同城推荐系统架构设计与实现
如何设计推荐系统的架构,是本文重点讨论的内容。
1968 0