def convert(self): images_pathes = self._get_images_pathes() dataset_name = os.path.basename(os.path.normpath(self.dataset_dir)) out_pr = ProjectStructure(self.settings['res_names']['project']) for image_fp in images_pathes: image_ext = os.path.splitext(image_fp)[1] image_name = os.path.splitext(image_fp)[0] dt = {"image_ext": image_ext} out_pr.add_item(dataset_name, image_name, dt) out_pr_fs = ProjectFS(self.out_dir, out_pr) out_pr_fs.make_dirs() res_meta = ProjectMeta() for class_name in self.classes: # TODO: Fix it line by new meta and object classes res_meta.classes.add({ 'title': class_name, 'shape': 'bitmap', 'color': color_utils.rgb2hex(color_utils.random_rgb()) }) res_meta.to_dir(out_pr_fs.project_path) progress = progress_counter.progress_counter_import( out_pr.name, out_pr.image_cnt) for sample_info in out_pr_fs: self._convert_sample(sample_info, res_meta) progress.iter_done_report()
def convert(self): images_pathes = self._get_images_pathes() masks_map = self._get_masks_mapping() dataset_name = 'ds' out_pr = ProjectStructure(self.settings['res_names']['project']) for image_fp in images_pathes: base_name = os.path.basename(image_fp) image_ext = os.path.splitext(image_fp)[1] image_id = os.path.splitext(base_name)[0] if base_name.replace(image_ext, '') in masks_map: dt = {"image_ext": ".png", "image_orig_path": image_fp} out_pr.add_item(dataset_name, image_id, dt) out_pr_fs = ProjectFS(self.out_dir, out_pr) out_pr_fs.make_dirs() res_meta = ProjectMeta() # TODO: Fix it line by new meta and object classes res_meta.classes.add({ 'title': 'leaf', 'shape': 'bitmap', 'color': color_utils.rgb2hex(color_utils.random_rgb()) }) res_meta.to_dir(out_pr_fs.project_path) progress = progress_counter.progress_counter_import( out_pr.name, out_pr.image_cnt) for sample_info in out_pr_fs: self._convert_sample(sample_info, masks_map, res_meta) progress.iter_done_report()