YOLO Prelabeling

Script for automated pre-labelling using YOLO

src.pipeline.prelabelling.yolo_prelabelling.generate_yolo_prelabelling(raw_dir: Path, output_dir: Path, model_path: Path, config: Dict, verbose: bool = False) → None

Generate predictions for all images in the raw directory using YOLO model.

Parameters:

• raw_dir (Path) – Path to directory containing raw images

• output_dir (Path) – Path to save prediction results

• model_path (Path) – Path to the YOLO model file

• config (Dict) – Configuration dictionary containing pipeline parameters