本篇是对测试集进行测试和相应结果的输出
比赛链接
https://www.tinymind.cn/competitions/47?rron=banner
输入信息
也就是我们的测试集
因为处理测试集的时候,我直接用的keras内置的图片生成器,所以数据需要手动处理一下。
将所有图片放入一个文件夹中,然后在旁边创建两个空文件夹就行,注意都用英文字母。
如下:
需要输出的信息
代码源码
from keras.preprocessing.image import ImageDataGenerator
from keras.models import load_model
testgen_dir = './val_data/'
#测试集目录
fin_list = [0.1,0.2,0.5,1,2,5,10,50,100]
#相应标签对应的人民币面值
def get_result(directory):
file = open('fin_data.csv','w')
file.write('name, label\n')
end = False # 相当于一个开关,当Ture的时候程序结束
i = 0 #记录测试的图片数目
#创建图片生成器
test_datagen = ImageDataGenerator(
rescale=1. / 255 #归一化
)
test_generator = test_datagen.flow_from_directory(
directory=directory, #地址