diff --git a/deep_sort/utils/parser.py b/deep_sort/utils/parser.py index 27fcf50..3dcd5bb 100644 --- a/deep_sort/utils/parser.py +++ b/deep_sort/utils/parser.py @@ -20,7 +20,7 @@ class YamlParser(edict): def merge_from_file(self, config_file): with open(config_file, 'r') as fo: - self.update(yaml.load(fo.read())) + self.update(yaml.load(fo.read(), Loader=yaml.CLoader)) def merge_from_dict(self, config_dict): @@ -35,4 +35,4 @@ if __name__ == "__main__": cfg = YamlParser(config_file="../configs/yolov3.yaml") cfg.merge_from_file("../configs/deep_sort.yaml") - import ipdb; ipdb.set_trace() \ No newline at end of file + import ipdb; ipdb.set_trace()