python+opencv讀取資料夾圖片並儲存
阿新 • • 發佈:2020-12-09
2020.12.7
紀念一下寫了一週的讀取資料夾圖片並儲存的指令碼
在pycharm裡面寫了一週啊,孩子真的寫哭了,,滿滿都是淚,今天換了vscode寫,他就對了!!!!
咱實在不知道為啥啊!!!!
那就先把正確程式碼儲存一下吧
-
先放資料夾結構
- -
然後是程式碼
import os
import cv2
'''
讀取資料夾下圖片並儲存到指定路徑從
'''
def read_path(file_pathname):
#遍歷該目錄下的所有圖片檔案
for filename in os.listdir(file_pathname):
print(filename)
img = cv2.imread(file_pathname+'/'+filename)
####change to gray
#(下面第一行是將RGB轉成單通道灰度圖,第二步是將單通道灰度圖轉成3通道灰度圖)不需要這種操作只需註釋掉即可
img=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
image_np=cv2.cvtColor(img,cv2.COLOR_GRAY2BGR)
#####save figure
cv2.imwrite('./MMAFEDB_dataset/save_test' +"/"+filename,image_np)
#注意*處如果包含家目錄(home)不能寫成~符號代替
#讀取的目錄
read_path("./MMAFEDB_dataset/input_test") # vscode裡面讀取圖片資料夾的正確方式,pycharm裡不知道。。。。
#print(os.getcwd())