Abhiram V
commited on
Update yolo.py with yaml.SafeLoader (#1970)
Browse files- models/yolo.py +1 -1
models/yolo.py
CHANGED
@@ -71,7 +71,7 @@ class Model(nn.Module):
|
|
71 |
import yaml # for torch hub
|
72 |
self.yaml_file = Path(cfg).name
|
73 |
with open(cfg) as f:
|
74 |
-
self.yaml = yaml.load(f, Loader=yaml.
|
75 |
|
76 |
# Define model
|
77 |
ch = self.yaml['ch'] = self.yaml.get('ch', ch) # input channels
|
|
|
71 |
import yaml # for torch hub
|
72 |
self.yaml_file = Path(cfg).name
|
73 |
with open(cfg) as f:
|
74 |
+
self.yaml = yaml.load(f, Loader=yaml.SafeLoader) # model dict
|
75 |
|
76 |
# Define model
|
77 |
ch = self.yaml['ch'] = self.yaml.get('ch', ch) # input channels
|