File size: 887 Bytes
5f1587b dc8325f 5f1587b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
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)
|