utils/config.py

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)