Update yolo.py with yaml.SafeLoader (#1970)
parent
91c30e4eff
commit
4501169547
|
@ -71,7 +71,7 @@ class Model(nn.Module):
|
|||
import yaml # for torch hub
|
||||
self.yaml_file = Path(cfg).name
|
||||
with open(cfg) as f:
|
||||
self.yaml = yaml.load(f, Loader=yaml.FullLoader) # model dict
|
||||
self.yaml = yaml.load(f, Loader=yaml.SafeLoader) # model dict
|
||||
|
||||
# Define model
|
||||
ch = self.yaml['ch'] = self.yaml.get('ch', ch) # input channels
|
||||
|
|
Loading…
Reference in New Issue