Unverified 提交 45011695 authored 作者: Abhiram V's avatar Abhiram V 提交者: GitHub

Update yolo.py with yaml.SafeLoader (#1970)

上级 91c30e4e
...@@ -71,7 +71,7 @@ class Model(nn.Module): ...@@ -71,7 +71,7 @@ class Model(nn.Module):
import yaml # for torch hub import yaml # for torch hub
self.yaml_file = Path(cfg).name self.yaml_file = Path(cfg).name
with open(cfg) as f: 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 # Define model
ch = self.yaml['ch'] = self.yaml.get('ch', ch) # input channels ch = self.yaml['ch'] = self.yaml.get('ch', ch) # input channels
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论