猫狗大战——基于TensorFlow的猫狗识别（2）

最新推荐文章于 2023-06-21 08:00:00 发布

VIP文章 i小玉

最新推荐文章于 2023-06-21 08:00:00 发布

阅读量3.3k

点赞数 7

分类专栏： project 文章标签： python 深度学习 tensorflow 人工智能

本文链接：https://blog.csdn.net/qq_33828738/article/details/105925465

版权

微信公众号：龙跃十二

我是小玉，一个平平无奇的小天才！

上篇文章我们说了关于猫狗大战这个项目的一些准备工作，接下来，我们看看具体的代码详解。

猫狗大战——基于TensorFlow的猫狗识别（1）

文件名：input_data.py

一、模块的导入：

import tensorflow as tf
import numpy as np
import os

tensorflow和numpy小玉就不在这里赘述了，os模块包含操作系统相关的功能，可以处理文件和目录这些我们日常手动需要做的操作。因为我们需要获取test目录下的文件，所以要导入os模块。

二、获取文件路径和标签

def get_files(file_dir):
    # file_dir: 文件夹路径
    # return: 乱序后的图片和标签

    cats = []
    label_cats = []
    dogs = []
    label_dogs = []
    # 载入数据路径并写入标签值
    for file in os.listdir(file_dir):
        name = file.split(sep='.')
        if name[0] == 'cat':
            cats.append(file_dir + file)
            label_cats.append(0)
        else:
            dogs.append(file_dir + file)
            label_dogs.append(1)
    print(