from .yacs import CfgNode cfg = CfgNode(new_allowed=True) cfg.save_dir = "./" cfg.pipeline = CfgNode(new_allowed=True) # Detector config cfg.pipeline.detector = CfgNode(new_allowed=True) cfg.pipeline.detector.model = "yolov8n" cfg.pipeline.detector.categories = [] cfg.pipeline.detector.thresholds = CfgNode(new_allowed=True) cfg.pipeline.detector.thresholds.confidence = 0.6 cfg.pipeline.detector.thresholds.iou = 0.4 cfg.pipeline.detector.slicing = CfgNode(new_allowed=True) cfg.pipeline.detector.slicing.overlap = 0.2 cfg.pipeline.detector.device = "cpu" # Tracker config cfg.pipeline.tracker = CfgNode(new_allowed=True) cfg.pipeline.tracker.algorithm = "dummytrack" def load_config(cfg, args_cfg): cfg.defrost() cfg.merge_from_file(args_cfg) cfg.freeze() if __name__ == "__main__": import sys with open(sys.argv[1], "w") as f: print(cfg, file=f)