Tensorflow -- 图像分类任务中读取自己的图片并添加标签

借用猫狗大战的图片来做数据读取的演示

import tensorflow as tf
import os
import numpy as np
import matplotlib.pyplot as plt

cat_filename = []
cat_label = []
dog_filename = []
dog_label = []
#tensorflow数据读取机制：包括创建文件名队列、读取文件放入文件队列中
#首先定义函数，将所有图片的文件名加入队列
def get_filename(file_dir):
    #获取猫和狗图片的地址
    for filename in os.listdir(file_dir +"/Cat"):
        cat_filename.append(file_dir +