Commit 58af91a2 authored by Barcis, Michal's avatar Barcis, Michal
Browse files

fixed imports

parent 6dd217a8
...@@ -18,14 +18,11 @@ from rclpy.qos import ( ...@@ -18,14 +18,11 @@ from rclpy.qos import (
QoSHistoryPolicy, QoSHistoryPolicy,
QoSReliabilityPolicy, QoSReliabilityPolicy,
) )
from rclpy.qos import QoSPresetProfiles
from mission_manager.client import MissionExecutor, MissionClient from mission_manager.client import MissionExecutor, MissionClient
from infdist.optimization.agent import FullCommAgent from infdist.optimization.agent import FullCommAgent
from infdist.robot.ros2_network import ROS2Network, ROS2NativeMessage from infdist.robot.ros2_network import ROS2Network, ROS2NativeMessage
from mission import generate_mission_context from .mission import generate_mission_context
from detect import detect_red
CUSTOM_QOS = QoSProfile( CUSTOM_QOS = QoSProfile(
history=QoSHistoryPolicy.RMW_QOS_POLICY_HISTORY_KEEP_LAST, history=QoSHistoryPolicy.RMW_QOS_POLICY_HISTORY_KEEP_LAST,
...@@ -203,7 +200,6 @@ class CameraSubscriber(Node, MissionExecutor): ...@@ -203,7 +200,6 @@ class CameraSubscriber(Node, MissionExecutor):
img = msg.data img = msg.data
npimg = np.frombuffer(img, dtype=np.uint8) npimg = np.frombuffer(img, dtype=np.uint8)
img = cv2.imdecode(npimg, cv2.IMREAD_COLOR)/255 img = cv2.imdecode(npimg, cv2.IMREAD_COLOR)/255
# mask = detect_red(img)
# mask = cv2.bitwise_not(mask) # mask = cv2.bitwise_not(mask)
# img = cv2.bitwise_and(img, img, mask=mask)/256 # img = cv2.bitwise_and(img, img, mask=mask)/256
host_id = self.hosts[host] host_id = self.hosts[host]
......
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
from infdist.optimization.models import MissionContext, InformationType from infdist.optimization.models import MissionContext, InformationType
from infdist.optimization.aggregations import AggregationMostRecent from infdist.optimization.aggregations import AggregationMostRecent
from utility import PeriodicTypeForecast, UtilityVideo from .utility import PeriodicTypeForecast, UtilityVideo
# from infdist.optimization.message_forecast import PeriodicTypeForecast # from infdist.optimization.message_forecast import PeriodicTypeForecast
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment