版權(quán)聲明：本文為博主原創(chuàng)文章，遵循 CC 4.0 BY-SA 版權(quán)協(xié)議，轉(zhuǎn)載請附上原文出處鏈接和本聲明。

**本文將為大家講述如何讀取多個文件夾下的圖片并貼標(biāo)簽（一個文件夾一個標(biāo)簽）
圖片存放方式：

目標(biāo)
讀取picture文件夾里兩個文件夾內(nèi)的所有圖片，并添加標(biāo)簽（為了方便起見每個文件夾放了三張圖片，分辨率為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("訓(xùn)練集讀取")'''讀取路徑下所有子文件夾中的圖片并存入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)

運行結(jié)果

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.]

總結(jié)

以上是生活随笔為你收集整理的python读取多个文件夹里的图片并添加标签的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。