计算机视觉(python)制作自己的图片分类数据集

最新推荐文章于 2024-06-10 18:41:51 发布

flyingingg

最新推荐文章于 2024-06-10 18:41:51 发布

阅读量5.1k

点赞数 6

分类专栏：计算机视觉文章标签： tensorflow 深度学习

本文链接：https://blog.csdn.net/Oscarouyangyafei/article/details/107163584

版权

本文介绍如何以简单易懂的方式创建自定义图片分类数据集，类似MNIST手写数字数据集。通过创建特定文件夹组织图像，利用TensorFlow 2.0进行训练集、验证集和测试集的划分，并将数据保存为npz文件。文章提供完整项目链接及后续数据加载方法。

摘要由CSDN通过智能技术生成

本文主要是记录自己用一种最通俗易懂的方法，生成类似调用minist手写数字数据集的方法喜欢记得关注我点收藏不迷路辛苦整理免费分享的

完整的项目地址在这里

首先把要进行分类的图像像下面一样分多少类就创建多少文件夹存放文件我这里对pokemon里的五种宠物小精灵进行分类

文件夹里面长这样

然后写代码进行训练集验证集测试集的分类并保存成npz格式方便读取

reminder：这里的tf是2.0版本的

import glob
import os 
import cv2
import numpy as np
import random
import tensorflow as tf
from tensorflow import keras

tf.random.set_seed(2222)
np.random.seed(2222)
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
assert tf.__version__.startswith('2.')

def Data_Generation():
    X_data=[];Y_data