Revert "feat(新功能): fix(修复bug): print 于 sketch拉伸导致的print比例不正确问题 docs(文档变更): refactor(重构): test(增加测试):"
This reverts commit bba91b46
This commit is contained in:
@@ -1,6 +1,9 @@
|
||||
import io
|
||||
import logging
|
||||
|
||||
import cv2
|
||||
import numpy as np
|
||||
from PIL import Image
|
||||
|
||||
from app.service.utils.new_oss_client import oss_get_image
|
||||
|
||||
@@ -35,18 +38,6 @@ class LoadImage:
|
||||
|
||||
def __call__(self, result):
|
||||
result['image'], result['pre_mask'] = self.read_image(result['path'])
|
||||
|
||||
# 判断是否resize sketch 保留ori image 用于模型输入
|
||||
result['ori_image'] = result['image']
|
||||
if result['resize_scale'][0] != 0 and result['resize_scale'][1] != 0:
|
||||
height, width = result['image'].shape[:2]
|
||||
# 计算新的宽度和高度
|
||||
new_width = int(width * result['resize_scale'][0])
|
||||
new_height = int(height * result['resize_scale'][1])
|
||||
# 使用cv2.resize()函数进行缩放
|
||||
result['image'] = cv2.resize(result['image'], (new_width, new_height))
|
||||
if result['pre_mask'] is not None:
|
||||
result['pre_mask'] = cv2.resize(result['pre_mask'], (new_width, new_height))
|
||||
result['gray'] = cv2.cvtColor(result['image'], cv2.COLOR_BGR2GRAY)
|
||||
result['keypoint'] = self.get_keypoint(result['name'])
|
||||
result['img_shape'] = result['image'].shape
|
||||
|
||||
Reference in New Issue
Block a user