From d66fcb33d79aaa097f3ffc21d95ebe7331f81d39 Mon Sep 17 00:00:00 2001 From: Benteng Ma Date: Fri, 23 Feb 2024 15:11:11 +0000 Subject: [PATCH] correct the lost module. --- .../lasr_vision_torch/src/lasr_vision_torch/__init__.py | 4 ++-- .../src/lasr_vision_torch/image_with_masks_and_attributes.py | 2 +- skills/src/lasr_skills/describe_people.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/common/vision/lasr_vision_torch/src/lasr_vision_torch/__init__.py b/common/vision/lasr_vision_torch/src/lasr_vision_torch/__init__.py index 1498265c2..5fcb001f7 100644 --- a/common/vision/lasr_vision_torch/src/lasr_vision_torch/__init__.py +++ b/common/vision/lasr_vision_torch/src/lasr_vision_torch/__init__.py @@ -3,8 +3,8 @@ from colour_estimation import load_images_to_dict, generate_colour_table, count_colours_in_masked_area, compare_colour_distributions from colour_estimation import SPESIFIC_COLOURS, DETAILED_COLOURS -from categories_and_attributes import CelebAMaskHQCategoriesAndAttributes -from image_with_masks_and_attributes import ImageWithMasksAndAttributes, ImageOfPerson +from lasr_vision_torch.categories_and_attributes import CategoriesAndAttributes, CelebAMaskHQCategoriesAndAttributes +from lasr_vision_torch.image_with_masks_and_attributes import ImageWithMasksAndAttributes, ImageOfPerson import numpy as np import cv2 diff --git a/common/vision/lasr_vision_torch/src/lasr_vision_torch/image_with_masks_and_attributes.py b/common/vision/lasr_vision_torch/src/lasr_vision_torch/image_with_masks_and_attributes.py index e244445de..b30724138 100644 --- a/common/vision/lasr_vision_torch/src/lasr_vision_torch/image_with_masks_and_attributes.py +++ b/common/vision/lasr_vision_torch/src/lasr_vision_torch/image_with_masks_and_attributes.py @@ -1,5 +1,5 @@ import numpy as np -from categories_and_attributes import CategoriesAndAttributes +from lasr_vision_torch.categories_and_attributes import CategoriesAndAttributes def _softmax(x: list[float]) -> list[float]: diff --git a/skills/src/lasr_skills/describe_people.py b/skills/src/lasr_skills/describe_people.py index 095d0f766..b30d63daa 100755 --- a/skills/src/lasr_skills/describe_people.py +++ b/skills/src/lasr_skills/describe_people.py @@ -141,7 +141,7 @@ def execute(self, userdata): xyz = userdata.xyz # xyz = np.nanmean(xyz, axis=2) # rospy.loginfo("COORD_Z:::%s" % str(xyz[neck_coord[0]][neck_coord[1]])) - point_head_client(xyz, neck_coord[0], neck_coord[1], client) + # point_head_client(xyz, neck_coord[0], neck_coord[1], client) return 'succeeded' except rospy.ServiceException as e: rospy.logwarn(f"Unable to perform inference. ({str(e)})")