利用tensorflow训练自己的图片数据（1）——预处理

最新推荐文章于 2025-07-14 17:41:07 发布

原创

最新推荐文章于 2025-07-14 17:41:07 发布 · 4.2w 阅读

263 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #图片 #数据 #标签

一. 准备原始数据

首先，我们需要准备训练的原始数据，本次训练为图像分类识别，因而一开始，笔者从网上随机的下载了Dog的四种类别：husky，jiwawa，poodle，qiutian。每种类别30种，一共120张图片。在训练之前，需要做的就是进行图像的预处理，即将这些大小不一的原始图片转换成我们训练需要的shape。

下载的原始图片分别放到同一文件的不同文件夹下，如：

二. 编程实现

该部分包括：制作Tfrecords，读取Tfrecords数据获得iamge和label，打印验证并保存生成的图片。

#将原始图片转换成需要的大小，并将其保存
#========================================================================================
import os  
import tensorflow as tf  
from PIL import Image  
  
#原始图片的存储位置
orig_picture = 'E:/train_test/train_data/generate_s