Add yolov5 P6 to hubconf.py

zhiqwang · Oct 24, 2021 · ca5c7f9 · ca5c7f9
1 parent c45b129
commit ca5c7f9
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -1,4 +1,13 @@
 # Optional list of dependencies required by the package
 dependencies = ["torch", "torchvision"]
 
-from yolort.models import yolov5n, yolov5s, yolov5m, yolov5l, yolov5ts
+from yolort.models import (
+    yolov5n,
+    yolov5n6,
+    yolov5s,
+    yolov5s6,
+    yolov5m,
+    yolov5m6,
+    yolov5l,
+    yolov5ts,
+)