图像分类之花卉图像分类（五）测试数据

这个时候测试集就要用到了，为了便于观察，我们这里先给测试集重命名，这样子，哪张图片分类错了，我们也比较好找。
重命名代码：

#!/usr/bin/python # -*- coding:utf-8 -*- import os outer_path = 'D:/flowers/test' folderlist = os.listdir(outer_path) # 列举文件夹 for folder in folderlist: inner_path = os.path.join(outer_path, folder) total_num_folder = len(folderlist) # 文件夹的总数 print('total have %d folders' % (total_num_folder)) filelist = os.listdir(inner_path) # 列举图片 i = 0 for item in filelist: total_num_file = len(filelist) # 单个文件夹内图片的总数 if item.endswith('.jpg'): src = os.path.join(os.path.abspath(inner_path), item) # 原图的地址 dst = os.path.join(os.path.abspath(inner_path), str(folder) + '_' + str( i) + '.jpg') # 新图的地址（这里可以把str(folder) + '_' + str(i) + '.jpg'改成你想改的名称） try: os.rename(src, dst) print 'converting %s to %s ...' % (src, dst) i += 1 except: continue print 'total %d to rename & converted %d jpgs' % (total_num_file, i)

随便打开测试集的，可以看到结果如下：

下面说下测试代码：
测试一整个文件夹的，这里的整个文件夹并不是指test文件夹，而是test文件夹下的五个子文件夹的任意一个，代码restall.py如下：

from PIL import Image import numpy as np import tensorflow as tf import matplotlib.pyplot as plt import model #from input_data import get_files import inp123456