visibility_threshold_dict={'bolt-roi': 0.01},
allow_unfound_seg=False,
class_merge_map={
'seg_mark-middle_0': 'seg_mark-inner_0',
#'seg_mark_2': 'seg_mark_0'
}
)
# mark_dataset.save_to_path('uncropped_mark.json', overwrite=True)
mark_dataset.display_preview(show_details=True, window_name='Mark')
# Mark (Cropped) Dataset Creation
logger.info('Creating Mark Dataset (Cropped Version)')
marker_dst_dir = f'{target_dst_dir}/marker'
make_dir_if_not_exists(marker_dst_dir)
cropped_mark_dataset = COCO_Dataset.new()
cropped_mark_dataset.categories = mark_categories.copy()
cropped_mark_dataset.licenses = mark_dataset.licenses.copy()
crop_pbar = tqdm(total=len(bolt_roi_dataset.images), unit='image(s)')
crop_pbar.set_description('Cropping')
for roi_image in bolt_roi_dataset.images:
orig_img = cv2.imread(roi_image.coco_url)
roi_anns = bolt_roi_dataset.annotations.get_annotations_from_imgIds([roi_image.id])
mark_images = mark_dataset.images.get_images_from_file_name(roi_image.file_name)
assert len(mark_images) == 1
mark_image = mark_images[0]
mark_anns = mark_dataset.annotations.get_annotations_from_imgIds(mark_image.id)
#assert len(roi_anns) == len(mark_anns)
for i, roi_ann in enumerate(roi_anns):
from annotation_utils.coco.structs import COCO_Dataset, \
COCO_License, COCO_Image, COCO_Annotation, COCO_Category
from common_utils.common_types.bbox import BBox
from logger import logger
dataset = COCO_Dataset.new(description='Test')
dataset.categories.append(
COCO_Category(id=len(dataset.categories),
supercategory='test_category',
name='category_a'))
dataset.categories.append(
COCO_Category(id=len(dataset.categories),
supercategory='test_category',
name='category_b'))
dataset.categories.append(
COCO_Category(id=len(dataset.categories),
supercategory='test_category',
name='category_c'))
for i in range(10):
dataset.licenses.append(
COCO_License(url=f'test_license_{i}',
name=f'Test License {i}',
id=len(dataset.licenses)))
for i in range(20):
dataset.images.append(
COCO_Image(license_id=i % len(dataset.licenses),
file_name=f'{i}.jpg',
coco_url=f'/path/to/{i}.jpg',
height=500,
width=500,