欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python+opencv读取文件夹图片并保存

程序员文章站 2022-04-09 13:08:52
...

2020.12.7

纪念一下写了一周的读取文件夹图片并保存的脚本

在pycharm里面写了一周啊,孩子真的写哭了,,满满都是泪,今天换了vscode写,他就对了!!!!
咱实在不知道为啥啊!!!!

那就先把正确代码保存一下吧

  • 先放文件夹结构
    -python+opencv读取文件夹图片并保存

  • 然后是代码

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())