feat(新功能):

fix(修复bug):  clothing seg 增加类型为sketch的预处理
docs(文档变更):
refactor(重构):
test(增加测试):
This commit is contained in:
zchengrong
2025-04-23 14:40:11 +08:00
parent b8fe29e735
commit a14e6051b1

View File

@@ -62,7 +62,7 @@ class ClothingSeg:
image = data["image"]
clothing_result = []
if image_type == "sketch":
seg_mask = get_seg_result(1, image)
seg_mask = get_seg_result(1, image[:, :, :3])
temp = seg_mask != 0.0
mask = (255 * (temp + 0).astype(np.uint8))
x_min, y_min, x_max, y_max = get_bounding_box(mask)