美文网首页
数据集从图片数据转换成numpy数组

数据集从图片数据转换成numpy数组

作者: dc_3 | 来源:发表于2019-09-26 09:32 被阅读0次
  • 首先从文件夹中读取'.jpg'后缀的所有图片名称
import glob
#TRAIN_IMG_PATH表示文件夹的路径
all_img_paths = glob.glob(os.path.join(TRAIN_IMG_PATH, "*.jpg"))

all_img_paths得到的是个list,大小=图片的总数。内容是每个图片的路径名称。
E:\deeplearningprogram\keraslearn\test0.1\train\c01.png
E:\deeplearningprogram\keraslearn\test0.1\train\c02.png
..........................................................................................
..........................................................................................
..........................................................................................
E:\deeplearningprogram\keraslearn\test0.1\train\s12.png
E:\deeplearningprogram\keraslearn\test0.1\train\s13.png

  • 使用cv.imread函数循环读取全部图片,归一化,存入到一个list里面
for img_path in all_img_paths:
    img = cv2.imread(img_path,0) # 以灰階讀入
    img = img.astype('float32')/255.
    imgs.append(img)
  • 将list转换成numpy数组
X = np.array(imgs)

相关文章

网友评论

      本文标题:数据集从图片数据转换成numpy数组

      本文链接:https://www.haomeiwen.com/subject/ugdguctx.html