Add P6 Backbones for YoloV5 (airctic#879)

* add p6 backbones

* fix path

Author: rsomani95

icevision/models/ultralytics/yolov5/backbones.py

@@ -1,17 +1,25 @@
 __all__ = [
     "small",
+    "small_p6",
     "medium",
+    "medium_p6",
     "large",
+    "large_p6",
     "extra_large",
+    "extra_large_p6",
 ]
 
 from icevision.models.ultralytics.yolov5.utils import *
 
 
 small = YoloV5BackboneConfig(model_name="yolov5s")
+small_p6 = YoloV5BackboneConfig(model_name="yolov5s6")
 
 medium = YoloV5BackboneConfig(model_name="yolov5m")
+medium_p6 = YoloV5BackboneConfig(model_name="yolov5m6")
 
 large = YoloV5BackboneConfig(model_name="yolov5l")
+large_p6 = YoloV5BackboneConfig(model_name="yolov5l6")
 
 extra_large = YoloV5BackboneConfig(model_name="yolov5x")
+extra_large_p6 = YoloV5BackboneConfig(model_name="yolov5x6")

icevision/models/ultralytics/yolov5/model.py

@@ -37,7 +37,11 @@ def model(
         else device
     )
 
-    cfg_filepath = Path(yolov5.__file__).parent / f"models/{model_name}.yaml"
+    if model_name in ["yolov5s", "yolov5m", "yolov5l", "yolov5x"]:
+        cfg_filepath = Path(yolov5.__file__).parent / f"models/{model_name}.yaml"
+    else:
+        cfg_filepath = Path(yolov5.__file__).parent / f"models/hub/{model_name}.yaml"
+
     if pretrained:
         weights_path = yolo_dir / f"{model_name}.pt"
 

tests/models/ultralytics/yolov5/test_model.py

@@ -5,7 +5,12 @@
 
 @pytest.mark.parametrize(
     "backbone",
-    [small, medium, large, extra_large],
+    # fmt: off
+    [
+        small,    medium,    large,    extra_large,
+        small_p6, medium_p6, large_p6, extra_large_p6,
+    ],
+    # fmt: on
 )
 def test_yolo_model(backbone):
     model = models.ultralytics.yolov5.model(