From 3dc0a8bddf52e495665e4ccd25726d1d0d3412bf Mon Sep 17 00:00:00 2001 From: Benteng Ma Date: Mon, 17 Jun 2024 17:41:22 +0100 Subject: [PATCH] Update attribute list for cloth --- .../image_with_masks_and_attributes.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/common/vision/lasr_vision_feature_extraction/src/lasr_vision_feature_extraction/image_with_masks_and_attributes.py b/common/vision/lasr_vision_feature_extraction/src/lasr_vision_feature_extraction/image_with_masks_and_attributes.py index 855260964..98637c30e 100644 --- a/common/vision/lasr_vision_feature_extraction/src/lasr_vision_feature_extraction/image_with_masks_and_attributes.py +++ b/common/vision/lasr_vision_feature_extraction/src/lasr_vision_feature_extraction/image_with_masks_and_attributes.py @@ -248,8 +248,11 @@ def describe(self) -> dict: print(self.attributes) for attribute in ['short sleeve top', 'long sleeve top', 'short sleeve outwear', - 'long sleeve outwear', 'shorts', - 'trousers', 'skirt', 'short sleeve dress', + 'long sleeve outwear', + # 'shorts', + # 'trousers', + # 'skirt', + 'short sleeve dress', 'long sleeve dress', 'vest dress', 'sling dress', 'sleeveless top']: result["attributes"][attribute] = False