最近在做深度学习计算机视觉的东西,需要创建自己的图片数据集(train,validation,test),并分别打包好生成npz文件,然后就可以像调用tf自带的mnist手写字数据集一样调用我的数据进行训练了,喜欢记得关注我 点收藏不迷路 辛苦整理免费分享的
这篇文主要记录一下如何把批量的图片打包到一个np数组中,这种方法速度挺快的,是我觉得目前最简单的创建数据集的方法
方法一:
创建list X_train=[], 采用X_train.append(img)批量加入图片,注意append完成后要将X_train 转化为np格式,X_train=np.array(X_train,dtype=float)
import glob
import os
import cv2
import numpy as np
# =============================================================================
# way1 创建list X_train=[]&