correct the lost module.

MBTMBTMBT · Feb 23, 2024 · d66fcb3 · d66fcb3
1 parent 3e384d1
commit d66fcb3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 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
@@ -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
@@ -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
@@ -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)})")