def __init__(self, config):
        WriterInterface.__init__(self, config)

        self.rgb_output_key = self.config.get_string("rgb_output_key",
                                                     "colors")
        self._supercategory = self.config.get_string("supercategory",
                                                     "coco_annotations")
        self.segmap_output_key = self.config.get_string(
            "segmap_output_key", "segmap")
        self.segcolormap_output_key = self.config.get_string(
            "segcolormap_output_key", "segcolormap")
        self.mask_encoding_format = self.config.get_string(
            "mask_encoding_format", "rle")
        self._append_to_existing_output = self.config.get_bool(
            "append_to_existing_output", False)
Ejemplo n.º 2
0
    def __init__(self, config):
        WriterInterface.__init__(self, config)

        # Parse configuration.
        self._dataset = self.config.get_string("dataset", "")

        self._append_to_existing_output = self.config.get_bool(
            "append_to_existing_output", False)

        # Save world to camera transformation
        self._save_world2cam = self.config.get_bool("save_world2cam", True)

        # Distance in meteres to object after which it is ignored. Mostly due to failed physics.
        self._ignore_dist_thres = self.config.get_float(
            "ignore_dist_thres", 100.)

        # Multiply the output depth image with this factor to get depth in mm.
        self._depth_scale = self.config.get_float("depth_scale", 1.0)

        # Output translation gt in mm
        self._mm2m = self.config.get_bool("m2mm", True)
Ejemplo n.º 3
0
 def __init__(self, config):
     WriterInterface.__init__(self, config)
     self.light_writer = ItemWriter(WriterUtility.get_light_attribute)
Ejemplo n.º 4
0
 def __init__(self, config):
     WriterInterface.__init__(self, config)
     self.object_writer = ItemWriter(WriterUtility.get_common_attribute)
Ejemplo n.º 5
0
 def __init__(self, config):
     WriterInterface.__init__(self, config)
     self._append_to_existing_output = self.config.get_bool(
         "append_to_existing_output", False)
     self._output_dir = self._determine_output_dir(False)
Ejemplo n.º 6
0
 def __init__(self, config):
     WriterInterface.__init__(self, config)
     self.object_writer = ItemWriter(WriterUtility._get_shapenet_attribute)