feat 处理多层sketch design出现的内搭越界问题

fix
This commit is contained in:
zhouchengrong
2024-07-22 15:41:07 +08:00
parent 87fcd5e9e9
commit 16c05c0f5b
2 changed files with 3 additions and 3 deletions

View File

@@ -3,7 +3,6 @@ import os
import numpy as np
from app.core.config import SEG_CACHE_PATH
from app.service.utils.decorator import ClassCallRunTime
from ..builder import PIPELINES
from ...utils.design_ensemble import get_seg_result
@@ -18,6 +17,7 @@ class Segmentation(object):
# @ClassCallRunTime
def __call__(self, result):
_, seg_result = self.load_seg_result(result["image_id"])
result['seg_result'] = seg_result
if not _:
result['seg_result'] = get_seg_result(result["image_id"], result['image'])
self.save_seg_result(result['seg_result'][0], result['image_id'])