From a3477f59d7e45ced4d1c5e551444b7fbab288268 Mon Sep 17 00:00:00 2001 From: sxj731533730 Date: Wed, 4 May 2022 15:24:13 +0800 Subject: [PATCH] Update parser.py --- deep_sort/utils/parser.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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()