版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

**本文将为大家讲述如何读取多个文件夹下的图片并贴标签（一个文件夹一个标签）
图片存放方式：

目标
读取picture文件夹里两个文件夹内的所有图片，并添加标签（为了方便起见每个文件夹放了三张图片，分辨率为3000*2000）

代码如下：

import numpy as np import cv2 import os from tqdm import tqdm label=np.zeros( (6) ) data_=np.zeros((6,512,512)) array_img = [] def image(path):print("-" * 50)print("训练集读取")'''读取路径下所有子文件夹中的图片并存入list'''train = []dir_counter = 0x=0i=0h=-1for child_dir in os.listdir(path):child_path = os.path.join(path, child_dir)h += 1for dir_image in tqdm(os.listdir(child_path)):img = cv2.imread(child_path + "\\" + dir_image, cv2.IMREAD_COLOR)img=cv2.resize(img,(512,512))img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)#灰度处理data=np.zeros((512,512))data[0:img.shape[0],0:img.shape[1]] = imgdata_[i, :, :] = data[0:512, 0:512]label[x] = hi += 1x += 1dir_counter += 1train.append(label)train.append(data_)return train image('picture') print('data_') print(data_) print('label') print(label)

运行结果

data_
[[[ 14. 15. 17. … 23. 25. 27.]
[ 13. 14. 17. … 25. 25. 26.]
[ 14. 14. 16. … 24. 27. 27.]
…
[144. 154. 162. … 146. 170. 164.]
[146. 119. 131. … 134. 172. 169.]
[176. 138. 142. … 117. 158. 168.]]

[[151. 152. 154. … 109. 104. 114.]
[156. 150. 154. … 105. 101. 110.]
[157. 152. 153. … 99. 104. 118.]
…
[ 52. 37. 32. … 144. 132. 134.]
[ 53. 45. 40. … 142. 136. 137.]
[ 54. 47. 47. … 138. 136. 141.]]

[[ 17. 17. 17. … 16. 16. 16.]
[ 18. 17. 17. … 17. 17. 17.]
[ 18. 17. 18. … 16. 16. 17.]
…
[ 17. 16. 16. … 131. 131. 132.]
[ 16. 17. 17. … 135. 133. 132.]
[ 17. 16. 17. … 136. 134. 133.]]

[[175. 170. 167. … 198. 198. 193.]
[171. 168. 170. … 193. 199. 190.]
[171. 168. 169. … 191. 191. 181.]
…
[147. 122. 130. … 158. 167. 161.]
[127. 106. 131. … 149. 136. 131.]
[139. 148. 141. … 150. 146. 122.]]

[[ 50. 52. 48. … 252. 244. 226.]
[ 43. 52. 50. … 225. 192. 171.]
[ 39. 43. 50. … 176. 138. 141.]
…
[117. 126. 140. … 64. 62. 56.]
[121. 119. 131. … 69. 70. 61.]
[110. 120. 125. … 85. 76. 71.]]

[[243. 246. 248. … 97. 95. 96.]
[242. 245. 248. … 99. 99. 94.]
[241. 244. 247. … 98. 96. 93.]
…
[ 19. 17. 17. … 54. 52. 54.]
[ 18. 17. 16. … 52. 53. 56.]
[ 17. 16. 16. … 50. 52. 54.]]]
label
[0. 0. 0. 1. 1. 1.]

总结

以上是生活随笔为你收集整理的python读取多个文件夹里的图片并添加标签的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。