Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

调整图片名

Browse Source
This commit is contained in:
liuyebo
2024-09-29 13:55:37 +08:00
parent 7a4cb5263a
commit a3fa1e502e
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
services/paddle_services/det_book.py
View File

@@ -17,10 +17,10 @@ def main():
img_path = request.form.get('img_path') img_path = request.form.get('img_path')
result = detector.get_book_areas(img_path) result = detector.get_book_areas(img_path)
dirname, img_name, ext = parse_img_path(img_path) dirname, img_name, img_ext = parse_img_path(img_path)
books_path = [] books_path = []
for i in range(len(result)): for i in range(len(result)):
save_path = os.path.join(dirname, img_name + '_book_' + str(i) + '.' + ext) save_path = os.path.join(dirname, f'{img_name}.book_{i}.{img_ext}')
cv2.imwrite(save_path, result[i]) cv2.imwrite(save_path, result[i])
books_path.append(save_path) books_path.append(save_path)
return books_path return books_path

4
services/paddle_services/dewarp.py
View File

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