码迷,mamicode.com
首页 > 编程语言 > 详细

Python如何读取指定文件夹下的所有图像

时间:2016-01-26 06:59:55      阅读:1601      评论:0      收藏:0      [点我收藏+]

标签:

(1)数据准备

数据集介绍:

数据集中存放的是1223幅图像,其中756个负样本(图像名称为0.1~0.756),458个正样本(图像名称为1.1~1.458),其中:"."前的标号为样本标签,"."后的标号为样本序号

(2)利用python读取文件夹中所有图像

 1 ‘‘‘
 2 Load the image files form the folder
 3 input:
 4     imgDir: the direction of the folder
 5     imgName:the name of the folder
 6 output:
 7     data:the data of the dataset
 8     label:the label of the datset
 9 ‘‘‘
10 def load_Img(imgDir,imgFoldName):
11     imgs = os.listdir(imgDir+imgFoldName)
12     imgNum = len(imgs)
13     data = np.empty((imgNum,1,12,12),dtype="float32")
14     label = np.empty((imgNum,),dtype="uint8")
15     for i in range (imgNum):
16         img = Image.open(imgDir+imgFoldName+"/"+imgs[i])
17         arr = np.asarray(img,dtype="float32")
18         data[i,:,:,:] = arr
19         label[i] = int(imgs[i].split(.)[0])
20     return data,label

调用方式

1 craterDir = "./data/CraterImg/Adjust/"
2 foldName = "East_CraterAdjust12"
3 data, label = load_Img(craterDir,foldName)

 

Python如何读取指定文件夹下的所有图像

标签:

原文地址:http://www.cnblogs.com/lutingting/p/5159193.html

(0)
(1)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!