如何从文件夹中读取所有图像并将同名图像保存在其他文件夹中python

Question

我的文件夹中有图像，我的文件名为 Doc0006.Row1City.jpg。以下代码在裁剪后以相同的名称保存图像，但我想像这样保存图像 Doc0006.Row1City_augment.jpg 我可以在其中更改此代码

import os.path
import glob
import cv2
def convertjpg(jpgfile,outdir):
    
    src = cv2.imread(jpgfile, cv2.IMREAD_UNCHANGED)
    
    try:
        dst = src[100:500, 200:600] 
        #dst = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
        print(dst.shape)
        cv2.imwrite(os.path.join(outdir,os.path.basename(jpgfile)), dst)
    except Exception as e:
        print(e)
        
for jpgfile in glob.glob(r'/home/hammad/thyroid/data/Hammad/Hammad/HH_Scans_Prediction/7104220_576_F/OriginalScan/*.png'):
    convertjpg(jpgfile,r"/home/hammad/thyroid/data/Hammad/Hammad/HH_Scans_Prediction/7104220_576_F/Cropped/")

Answer 1

您可以更改代码中的这一行：

cv2.imwrite(os.path.join(outdir,os.path.basename(jpgfile)), dst)

来自

cv2.imwrite(os.path.join(outdir,os.path.basename(jpgfile).replace('.jpg', '_augment.jpg')), dst)

Answer 2

您可以试试下面的代码：

cv2.imwrite(os.path.join(outdir,os.path.basename(jpgfile).replace(".jpg", "_augment.jpg")), dst)

解释：您提到的保存文件的默认名称是 Doc0006.Row1City.jpg。现在 replace 函数会将文件名的 .jpg 部分替换为 _augment.jpg。这将使用所需名称 Doc0006.Row1City_augment.jpg.

保存文件

如何从文件夹中读取所有图像并将同名图像保存在其他文件夹中python

How to read all image from folder and save image with same name in other folder python

python

opencv

glob

image-processing

python-3.x