fix      seg 缓存逻辑新增 size判断 避免出现seg缓存与图片大小不一致问题
This commit is contained in:
zhouchengrong
2024-10-04 17:37:42 +08:00
parent 9b415fc502
commit 571a84ac68

View File

@@ -36,7 +36,8 @@ class Segmentation:
# 本地查询seg 缓存是否存在
_, seg_result = self.load_seg_result(result["image_id"])
result['seg_result'] = seg_result
if not _:
# 判断缓存和实际图片size是否相同
if not _ or result["image"].shape[:2] != seg_result.shape:
# 推理获得seg 结果
seg_result = get_seg_result(result["image_id"], result['image'])[0]
self.save_seg_result(seg_result, result['image_id'])