|
|
from .yacs import CfgNode |
|
|
|
|
|
cfg = CfgNode(new_allowed=True) |
|
|
cfg.save_dir = "./" |
|
|
cfg.pipeline = CfgNode(new_allowed=True) |
|
|
|
|
|
|
|
|
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" |
|
|
|
|
|
|
|
|
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) |
|
|
|
