diff --git a/app/service/design_fast/item.py b/app/service/design_fast/item.py index 5ca89ab..3bee6bb 100644 --- a/app/service/design_fast/item.py +++ b/app/service/design_fast/item.py @@ -17,6 +17,8 @@ class OthersItem(BaseItem): LoadImage(minio_client), Segmentation(minio_client), Color(minio_client), + NoSegPrintPainting(minio_client), + PrintPainting(minio_client), Scaling(), Split(minio_client) ] diff --git a/app/service/design_fast/pipeline/segmentation.py b/app/service/design_fast/pipeline/segmentation.py index ebdb6c6..9468556 100644 --- a/app/service/design_fast/pipeline/segmentation.py +++ b/app/service/design_fast/pipeline/segmentation.py @@ -51,6 +51,8 @@ class Segmentation: if not _ or result["image"].shape[:2] != seg_result.shape: # 推理获得seg 结果 seg_result = get_seg_result(result['image']) + if result['name'] == 'others': + seg_result = seg_result.clip(max=1) self.save_seg_result(seg_result, result['image_id']) result['seg_result'] = seg_result