使用Python OpenCV模块编辑后，保存相同名称的图像

Question

我正在尝试在OpenCV编辑后，以相同的名称保存图片。我可以使用相同的名称保存。但无法保存到不同的文件中。因此，这是我的代码：

import cv2
import numpy as np
import glob

filename = [img for img in glob.glob("./mypath_1/*.jpg")]
flist=sorted(filename)
images = []
path='./mypath_2/'

for image in flist:
    img= cv2.imread(image)
    alpha=2
    beta=-420

    img2=cv2.addWeighted(img,alpha,np.zeros(img.shape,img.dtype),0,beta)
    hsv = cv2.cvtColor(img2, cv2.COLOR_BGR2HSV)
    cv2.imwrite( path+image, hsv)

此外，我尝试了以下代码：cv2.imwrite( './mypath_2/'+image, hsv)。

我没有保存图像，也没有在这段代码中收到错误消息。

有什么建议吗？

- Curious G.

Answer 1

import numpy as np
import cv2
import os
import sys
from pathlib import Path

if __name__ == "__main__":
    #get alpha and beta values
    alpha, beta =2, -420
    # get directory path where the images are stored
    image_dir = "/path/to/image read directory/"
    # get directory path where you want to save the images
    output_dir = "/path/to/image write directory/"
    #iterate through all the files in the image directory
    for _, _, image_names in os.walk(image_dir):
        #iterate through all the files in the image_dir
        for image_name in image_names:
            # check for extension .jpg
            if '.jpg' in image_name:
                # get image read path(path should not contain spaces in them)
                filepath = os.path.join(image_dir, image_name)
                # get image write path
                dstpath = os.path.join(output_dir, image_name)
                print(filepath, dstpath)
                # read the image
                image = cv2.imread(filepath)
                # do your processing
                image2 = cv2.addWeighted(image, alpha,np.zeros_like(image),0,beta)
                hsv = cv2.cvtColor(image2, cv2.COLOR_BGR2HSV)
                # write the image in a different path with the same name
                cv2.imwrite(dstpath, hsv)
                print(image.shape, image2.shape, hsv.shape)

- Operator77

Answer 2

你可以使用 nipath 三行代码来完成它。

import ntpath
filename = ntpath.basename("D:\coding\python\Sample Photos\1.jpg")
cv2.imwrite("D:\coding\python\Sample Photos\"+filename, img)

- Harshith J Poojary