fixed imports

camera_streaming/camera_subscriber.py

@@ -18,14 +18,11 @@ from rclpy.qos import (
     QoSHistoryPolicy,
     QoSReliabilityPolicy,
 )
-from rclpy.qos import QoSPresetProfiles
 
 from mission_manager.client import MissionExecutor, MissionClient
 from infdist.optimization.agent import FullCommAgent
 from infdist.robot.ros2_network import ROS2Network, ROS2NativeMessage
-from mission import generate_mission_context
-
-from detect import detect_red
+from .mission import generate_mission_context
 
 CUSTOM_QOS = QoSProfile(
     history=QoSHistoryPolicy.RMW_QOS_POLICY_HISTORY_KEEP_LAST,
@@ -203,7 +200,6 @@ class CameraSubscriber(Node, MissionExecutor):
             img = msg.data
             npimg = np.frombuffer(img, dtype=np.uint8)
             img = cv2.imdecode(npimg, cv2.IMREAD_COLOR)/255
-            # mask = detect_red(img)
             # mask = cv2.bitwise_not(mask)
             # img = cv2.bitwise_and(img, img, mask=mask)/256
             host_id = self.hosts[host]

camera_streaming/mission.py

@@ -3,7 +3,7 @@
 from infdist.optimization.models import MissionContext, InformationType
 from infdist.optimization.aggregations import AggregationMostRecent
 
-from utility import PeriodicTypeForecast, UtilityVideo
+from .utility import PeriodicTypeForecast, UtilityVideo
 # from infdist.optimization.message_forecast import PeriodicTypeForecast