python+opencv读取文件夹图片并保存
程序员文章站
2022-04-09 13:08:52
...
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())