aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat（新功能）:

Browse Source 
fix（修复bug）:
docs（文档变更）:
refactor（重构）:
test(增加测试):

代码回溯
This commit is contained in:
zhouchengrong
2025-02-07 16:12:04 +08:00
parent 78c6da3c7b
commit a6c237072e
3 changed files with 13 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
app/service/design_fast/pipeline/segmentation.py
View File

@@ -36,27 +36,12 @@ class Segmentation:
# preview 过模型 不缓存
if "preview_submit" in result.keys() and result['preview_submit'] == "preview":
# 推理获得seg 结果
seg_result = get_seg_result(result["image_id"], result['ori_image'])
if result['resize_scale'][0] != 0 and result['resize_scale'][1] != 0:
height, width = seg_result.shape[:2]
# 计算新的宽度和高度
new_width = int(width * result['resize_scale'][0])
new_height = int(height * result['resize_scale'][1])
# 使用cv2.resize()函数进行缩放
seg_result = cv2.resize(seg_result, (new_width, new_height))
seg_result = get_seg_result(result["image_id"], result['image'])
# submit 过模型 缓存
elif "preview_submit" in result.keys() and result['preview_submit'] == "submit":
# 推理获得seg 结果
seg_result = get_seg_result(result["image_id"], result['ori_image'])
seg_result_save = seg_result
if result['resize_scale'][0] != 0 and result['resize_scale'][1] != 0:
height, width = seg_result.shape[:2]
# 计算新的宽度和高度
new_width = int(width * result['resize_scale'][0])
new_height = int(height * result['resize_scale'][1])
# 使用cv2.resize()函数进行缩放
seg_result = cv2.resize(seg_result, (new_width, new_height))
self.save_seg_result(seg_result_save, result['image_id'])
seg_result = get_seg_result(result["image_id"], result['image'])
self.save_seg_result(seg_result, result['image_id'])
# null 正常流程 加载本地缓存 无缓存则过模型
else:
# 本地查询seg 缓存是否存在
@@ -64,16 +49,8 @@ class Segmentation:
# 判断缓存和实际图片size是否相同
if not _ or result["image"].shape[:2] != seg_result.shape:
# 推理获得seg 结果
seg_result = get_seg_result(result["image_id"], result['ori_image'])
seg_result_save = seg_result
if result['resize_scale'][0] != 0 and result['resize_scale'][1] != 0:
height, width = seg_result.shape[:2]
# 计算新的宽度和高度
new_width = int(width * result['resize_scale'][0])
new_height = int(height * result['resize_scale'][1])
# 使用cv2.resize()函数进行缩放
seg_result = cv2.resize(seg_result, (new_width, new_height))
self.save_seg_result(seg_result_save, result['image_id'])
seg_result = get_seg_result(result["image_id"], result['image'])
self.save_seg_result(seg_result, result['image_id'])
result['seg_result'] = seg_result
# 处理前片后片