Add always_publish arg

python/coral_usb/detector_base.py

@@ -131,9 +131,6 @@ def image_cb(self, msg):
             label_names=[self.label_names[lbl] for lbl in labels],
             label_proba=scores)
 
-        self.pub_rects.publish(rect_msg)
-        self.pub_class.publish(cls_msg)
-
         if self.enable_visualization:
             with self.lock:
                 self.img = img
@@ -143,6 +140,11 @@ def image_cb(self, msg):
                 self.labels = labels
                 self.scores = scores
 
+        if not self.always_publish and len(cls_msg.labels) <= 0:
+            return
+        self.pub_rects.publish(rect_msg)
+        self.pub_class.publish(cls_msg)
+
     def visualize_cb(self, event):
         if (not self.visualize or self.img is None or self.encoding is None
                 or self.header is None or self.bboxes is None

python/coral_usb/human_pose_estimator.py

@@ -191,6 +191,8 @@ def image_cb(self, msg):
             label_proba=[np.average(score) for score in scores]
         )
 
+        if not self.always_publish and len(cls_msg.label_names) <= 0:
+            return
         self.pub_pose.publish(poses_msg)
         self.pub_rects.publish(rects_msg)
         self.pub_class.publish(cls_msg)

python/coral_usb/node_base.py

@@ -124,6 +124,11 @@ def __init__(self, model_file=None, label_file=None, namespace='~'):
             self.timer = rospy.Timer(
                 rospy.Duration(self.duration), self.visualize_cb)
 
+        self.always_publish = rospy.get_param("~always_publish", True)
+        rospy.loginfo(
+            "Publish even if object/human_pose is not found : {}".format(
+                self.always_publish))
+
     def subscribe(self):
         if self.transport_hint == 'compressed':
             self.sub_image = rospy.Subscriber(