feat(新功能):
fix(修复bug): clothing seg 增加类型为sketch的预处理 docs(文档变更): refactor(重构): test(增加测试):
This commit is contained in:
@@ -62,7 +62,7 @@ class ClothingSeg:
|
|||||||
image = data["image"]
|
image = data["image"]
|
||||||
clothing_result = []
|
clothing_result = []
|
||||||
if image_type == "sketch":
|
if image_type == "sketch":
|
||||||
seg_mask = get_seg_result(1, image)
|
seg_mask = get_seg_result(1, image[:, :, :3])
|
||||||
temp = seg_mask != 0.0
|
temp = seg_mask != 0.0
|
||||||
mask = (255 * (temp + 0).astype(np.uint8))
|
mask = (255 * (temp + 0).astype(np.uint8))
|
||||||
x_min, y_min, x_max, y_max = get_bounding_box(mask)
|
x_min, y_min, x_max, y_max = get_bounding_box(mask)
|
||||||
|
|||||||
Reference in New Issue
Block a user