fix  design pipeline 剔除边缘检测任务,直接用分割
This commit is contained in:
zhouchengrong
2024-07-24 15:21:06 +08:00
parent b7e9131cf7
commit 8363ec9ab3
4 changed files with 15 additions and 29 deletions

View File

@@ -9,8 +9,8 @@ class Top(Clothing):
pipeline = [
dict(type='LoadImageFromFile', path=kwargs['path'], color=kwargs['color'], print_dict=kwargs['print']),
dict(type='KeypointDetection'),
dict(type='ContourDetection'),
dict(type='Segmentation', device='cpu', show=False, debug=kwargs['debug']),
# dict(type='ContourDetection'),
dict(type='Segmentation'),
dict(type='Painting', painting_flag=True),
dict(type='PrintPainting', print_flag=True),
# dict(type='ImageShow', key=['image', 'mask', 'seg_visualize', 'pattern_image']),