Object Detection
|
Functions | |
generate_synthetic_object_annotations (model_path, img_dir, output_dir, split, imgsz=320) | |
smooth_mask (binary_mask) | |
thin_mask (binary_mask) | |
process_multi_class_mask (mask_path, class_id_map={1:12, 2:3}, orig_width=320, orig_height=320, padded_size=320) | |
process_image (mask_path=None) | |
write_yolo_annotations (output_path, image_name, objects_info) | |
process_directory (mask_dir, output_dir=None) | |
merge_annotations (output_dir, lane_label_dir, da_dir, object_label_dir=None, split='') | |
main () | |
create_annotations.generate_synthetic_object_annotations | ( | model_path, | |
img_dir, | |||
output_dir, | |||
split, | |||
imgsz = 320 |
|||
) |
create_annotations.main | ( | ) |
create_annotations.merge_annotations | ( | output_dir, | |
lane_label_dir, | |||
da_dir, | |||
object_label_dir = None , |
|||
split = '' |
|||
) |
create_annotations.process_directory | ( | mask_dir, | |
output_dir = None |
|||
) |
create_annotations.process_image | ( | mask_path = None | ) |
create_annotations.process_multi_class_mask | ( | mask_path, | |
class_id_map = {1: 12, 2: 3} , |
|||
orig_width = 320 , |
|||
orig_height = 320 , |
|||
padded_size = 320 |
|||
) |
create_annotations.smooth_mask | ( | binary_mask | ) |
create_annotations.thin_mask | ( | binary_mask | ) |
create_annotations.write_yolo_annotations | ( | output_path, | |
image_name, | |||
objects_info | |||
) |