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()