diff --git a/models/yolo.py b/models/yolo.py index 5dc8b57f4d98..6cf9dde08d1a 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -71,7 +71,7 @@ def __init__(self, cfg='yolov5s.yaml', ch=3, nc=None): # model, input channels, 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