调整图片名

This commit is contained in:
2024-09-29 13:55:37 +08:00
parent 7a4cb5263a
commit a3fa1e502e
2 changed files with 4 additions and 4 deletions

View File

@@ -17,8 +17,8 @@ def main():
img_path = request.form.get('img_path')
img = cv2.imread(img_path)
dewarped_img = dewarper.dewarp_image(img)
dirname, img_name, ext = parse_img_path(img_path)
save_path = os.path.join(dirname, img_name + '_dewarped.' + ext)
dirname, img_name, img_ext = parse_img_path(img_path)
save_path = os.path.join(dirname, f'{img_name}.dewarped.{img_ext}')
cv2.imwrite(save_path, dewarped_img)
return save_path