Object Detection
Loading...
Searching...
No Matches
Functions
create_annotations Namespace Reference

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

Function Documentation

◆ generate_synthetic_object_annotations()

create_annotations.generate_synthetic_object_annotations (   model_path,
  img_dir,
  output_dir,
  split,
  imgsz = 320 
)

◆ main()

create_annotations.main ( )

◆ merge_annotations()

create_annotations.merge_annotations (   output_dir,
  lane_label_dir,
  da_dir,
  object_label_dir = None,
  split = '' 
)

◆ process_directory()

create_annotations.process_directory (   mask_dir,
  output_dir = None 
)

◆ process_image()

create_annotations.process_image (   mask_path = None)

◆ process_multi_class_mask()

create_annotations.process_multi_class_mask (   mask_path,
  class_id_map = {1: 12, 2: 3},
  orig_width = 320,
  orig_height = 320,
  padded_size = 320 
)

◆ smooth_mask()

create_annotations.smooth_mask (   binary_mask)

◆ thin_mask()

create_annotations.thin_mask (   binary_mask)

◆ write_yolo_annotations()

create_annotations.write_yolo_annotations (   output_path,
  image_name,
  objects_info 
)