From 997ba7b346beede15009b187b6d1d47367f94434 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 5 Jul 2020 12:50:04 -0700
Subject: [PATCH] import yaml in yolo.py

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

diff --git a/models/yolo.py b/models/yolo.py
index 7961fab..b2d09cc 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -48,6 +48,7 @@ class Model(nn.Module):
         if type(model_cfg) is dict:
             self.md = model_cfg  # model dict
         else:  # is *.yaml
+            import yaml
             with open(model_cfg) as f:
                 self.md = yaml.load(f, Loader=yaml.FullLoader)  # model dict