tensorflow框架的猫狗大战代码及详解.zip

这是利用卷积神经网络进行猫和狗的图像分类的相关代码，由于训练和测试的数据集较大，所以没有放上来，训练数据集需要自行下载

import tensorflow as tf
import numpy as np
import os
‘‘‘生成图片路径和标签的List‘‘‘
def get_files（file_dir）:
    “““
       输入： 存放训练照片的文件地址
       返回:  图像列表， 标签列表
       “““
    cats = []
    label_cats = []
    dogs = []
    label_dogs = []
    for file in os.listdir（file_dir）:  # 用os.listdir函数来返回file_dir路径下所有的图片，file就是要读取的照片
        name = file.split（sep=‘.‘）  # 用file.split 将文件的名字分隔开，分隔的符号是‘.’
        if name[0] == ‘cat‘:  # 所以只用读取 . 前面这个字符串
            cats.append（file_dir + file）
            label_cats.append（0）  # 把图像和标签加入列表
        else:
            dogs.append（file_dir + file）
            label_dogs.append（1）  # 名字是cat就赋值0，名字是dog就赋值1
    print（‘There are %d cats\nThere are %d dogs‘ % （len（cats） len（dogs）））#打印有多少猫，多少狗
    ‘‘‘对生成的图片路径和标签List做打乱处理‘‘‘
    image_list = np.h

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     文件        5327  2019-10-21 16:55  代码\input_data.py
     文件       10635  2019-10-21 09:34  代码\model.py
     文件        2679  2019-10-21 17:06  代码\test.py
     文件        3677  2019-10-21 16:41  代码\training.py
     目录           0  2020-03-19 10:17  代码\