Add ultralytics like model loading notebooks (#52)

* Add YOLO to module * Add ultralytics like model loading notebooks
zhiqwang · Feb 5, 2021 · 38bdf06 · 38bdf06
1 parent 1c5ff1e
commit 38bdf06
Show file tree

Hide file tree

Showing 2 changed files with 507 additions and 1 deletion.
diff --git a/models/yolo.py b/models/yolo.py
@@ -13,7 +13,7 @@
 
 from typing import Tuple, Any, List, Dict, Optional
 
-__all__ = ['yolov5_darknet_pan_s_r31', 'yolov5_darknet_pan_m_r31', 'yolov5_darknet_pan_l_r31',
+__all__ = ['YOLO', 'yolov5_darknet_pan_s_r31', 'yolov5_darknet_pan_m_r31', 'yolov5_darknet_pan_l_r31',
            'yolov5_darknet_pan_s_r40', 'yolov5_darknet_pan_m_r40', 'yolov5_darknet_pan_l_r40']
 
 

diff --git a/notebooks/load-model-as-ultralytics.ipynb b/notebooks/load-model-as-ultralytics.ipynb