From fc697c7df88a41623fe2d8591232ab8de3b397dd Mon Sep 17 00:00:00 2001
From: Abhiram V <61599526+Anon-Artist@users.noreply.github.com>
Date: Tue, 19 Jan 2021 00:17:00 +0530
Subject: [PATCH] Update yolo.py with yaml.SafeLoader (#1970)

---
 models/yolo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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