Another saving type LabelImage

jsk-ros-pkg · Jul 21, 2016 · 43deb64 · 43deb64
1 parent 7c7c8d2
commit 43deb64
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/jsk_data/node_scripts/data_collection_server.py b/jsk_data/node_scripts/data_collection_server.py
@@ -109,6 +109,10 @@ def service_cb(self, req):
                 depth = bridge.imgmsg_to_cv2(msg)
                 with open(osp.join(save_dir, topic['fname']), 'wb') as f:
                     pkl.dump(depth, f)
+            elif topic['savetype'] == 'LabelImage':
+                bridge = cv_bridge.CvBridge()
+                label = bridge.imgmsg_to_cv2(msg)
+                cv2.imwrite(osp.join(save_dir, topic['fname']), label)
             else:
                 rospy.logerr('Unexpected savetype for topic: {}'
                              .format(topic['savetype']))