如何从文件夹中读取所有图像，并将相同名称的图像保存在其他文件夹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).replace(".jpg", "_augment.jpg")), dst)

说明:您所提到的保存的默认名称是Doc0006.Row1City.jpg。现在，replace函数将用_augment.jpg替换文件名中的.jpg部分。这将以所需的名称Doc0006.Row1City_augment.jpg保存文件。

Answer 2

您可以在代码中更改这一行:

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)