Skip to content

Commit 9855f08

Browse files
authored
Update test_2label.py
1 parent a3bbfb4 commit 9855f08

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

reid_eval/test_2label.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def get_id(img_path):
179179
###load config###
180180
config_path = os.path.join('../outputs',name,'config.yaml')
181181
with open(config_path, 'r') as stream:
182-
config = yaml.load(stream)
182+
config = yaml.safe_load(stream)
183183

184184
model_structure = ft_netAB(config['ID_class'], norm=config['norm_id'], stride=config['ID_stride'], pool=config['pool'])
185185

0 commit comments

Comments
 (0)