猫狗大战使用CNN分类

qixiao79134 23 0 PDF 2021-02-28 05:02:29

一.构造数据集 我们直接下载下来的猫狗数据集是分为训练集和测试集的25000张猫狗图像。 这个数据集太大了而且猫狗没有分类,我们需要创建一个新的数据集。分为每个类别各1000个样本的训练集和每个类别各500个样本的测试集,验证集。 import os,shutil #解压该目录的文件(训练集照片,25000猫狗图像) original_dataset_dir = 'G:\\Astudy\\python深度学习基于tensorflow\\dataset\\kaggle\\train' #在指定路径下创建文件夹 base_dir = 'G:\\Astudy\\python深度学习基于tenso

用户评论
请输入评论内容
评分:
暂无评论