aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

4
app/service/design/items/pipelines/split.py
View File

@@ -28,9 +28,9 @@ class Split(object):
result['front_mask'] = result['mask'].copy()
result['back_mask'] = np.zeros_like(result['mask'])
else:
temp_front = result['seg_result'] == 1
temp_front = result['seg_result'] == 1.0
result['front_mask'] = (result['mask'] * (temp_front + 0).astype(np.uint8))
temp_back = result['seg_result'] == 2
temp_back = result['seg_result'] == 2.0
result['back_mask'] = (result['mask'] * (temp_back + 0).astype(np.uint8))
if result['name'] in ('outwear', 'dress', 'blouse', 'skirt', 'trousers', 'tops', 'bottoms'):