bplusplus 1.1.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

bplusplus/yolov5detect/models/experimental.py

@@ -1,130 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-"""Experimental modules."""
-
-import math
-
-import numpy as np
-import torch
-import torch.nn as nn
-
-from utils.downloads import attempt_download
-
-
-class Sum(nn.Module):
-    """Weighted sum of 2 or more layers https://arxiv.org/abs/1911.09070."""
-
-    def __init__(self, n, weight=False):
-        """Initializes a module to sum outputs of layers with number of inputs `n` and optional weighting, supporting 2+
-        inputs.
-        """
-        super().__init__()
-        self.weight = weight  # apply weights boolean
-        self.iter = range(n - 1)  # iter object
-        if weight:
-            self.w = nn.Parameter(-torch.arange(1.0, n) / 2, requires_grad=True)  # layer weights
-
-    def forward(self, x):
-        """Processes input through a customizable weighted sum of `n` inputs, optionally applying learned weights."""
-        y = x[0]  # no weight
-        if self.weight:
-            w = torch.sigmoid(self.w) * 2
-            for i in self.iter:
-                y = y + x[i + 1] * w[i]
-        else:
-            for i in self.iter:
-                y = y + x[i + 1]
-        return y
-
-
-class MixConv2d(nn.Module):
-    """Mixed Depth-wise Conv https://arxiv.org/abs/1907.09595."""
-
-    def __init__(self, c1, c2, k=(1, 3), s=1, equal_ch=True):
-        """Initializes MixConv2d with mixed depth-wise convolutional layers, taking input and output channels (c1, c2),
-        kernel sizes (k), stride (s), and channel distribution strategy (equal_ch).
-        """
-        super().__init__()
-        n = len(k)  # number of convolutions
-        if equal_ch:  # equal c_ per group
-            i = torch.linspace(0, n - 1e-6, c2).floor()  # c2 indices
-            c_ = [(i == g).sum() for g in range(n)]  # intermediate channels
-        else:  # equal weight.numel() per group
-            b = [c2] + [0] * n
-            a = np.eye(n + 1, n, k=-1)
-            a -= np.roll(a, 1, axis=1)
-            a *= np.array(k) ** 2
-            a[0] = 1
-            c_ = np.linalg.lstsq(a, b, rcond=None)[0].round()  # solve for equal weight indices, ax = b
-
-        self.m = nn.ModuleList(
-            [nn.Conv2d(c1, int(c_), k, s, k // 2, groups=math.gcd(c1, int(c_)), bias=False) for k, c_ in zip(k, c_)]
-        )
-        self.bn = nn.BatchNorm2d(c2)
-        self.act = nn.SiLU()
-
-    def forward(self, x):
-        """Performs forward pass by applying SiLU activation on batch-normalized concatenated convolutional layer
-        outputs.
-        """
-        return self.act(self.bn(torch.cat([m(x) for m in self.m], 1)))
-
-
-class Ensemble(nn.ModuleList):
-    """Ensemble of models."""
-
-    def __init__(self):
-        """Initializes an ensemble of models to be used for aggregated predictions."""
-        super().__init__()
-
-    def forward(self, x, augment=False, profile=False, visualize=False):
-        """Performs forward pass aggregating outputs from an ensemble of models.."""
-        y = [module(x, augment, profile, visualize)[0] for module in self]
-        # y = torch.stack(y).max(0)[0]  # max ensemble
-        # y = torch.stack(y).mean(0)  # mean ensemble
-        y = torch.cat(y, 1)  # nms ensemble
-        return y, None  # inference, train output
-
-
-def attempt_load(weights, device=None, inplace=True, fuse=True):
-    """
-    Loads and fuses an ensemble or single YOLOv5 model from weights, handling device placement and model adjustments.
-
-    Example inputs: weights=[a,b,c] or a single model weights=[a] or weights=a.
-    """
-    from models.yolo import Detect, Model
-
-    model = Ensemble()
-    for w in weights if isinstance(weights, list) else [weights]:
-        ckpt = torch.load(attempt_download(w), map_location="cpu")  # load
-        ckpt = (ckpt.get("ema") or ckpt["model"]).to(device).float()  # FP32 model
-
-        # Model compatibility updates
-        if not hasattr(ckpt, "stride"):
-            ckpt.stride = torch.tensor([32.0])
-        if hasattr(ckpt, "names") and isinstance(ckpt.names, (list, tuple)):
-            ckpt.names = dict(enumerate(ckpt.names))  # convert to dict
-
-        model.append(ckpt.fuse().eval() if fuse and hasattr(ckpt, "fuse") else ckpt.eval())  # model in eval mode
-
-    # Module updates
-    for m in model.modules():
-        t = type(m)
-        if t in (nn.Hardswish, nn.LeakyReLU, nn.ReLU, nn.ReLU6, nn.SiLU, Detect, Model):
-            m.inplace = inplace
-            if t is Detect and not isinstance(m.anchor_grid, list):
-                delattr(m, "anchor_grid")
-                setattr(m, "anchor_grid", [torch.zeros(1)] * m.nl)
-        elif t is nn.Upsample and not hasattr(m, "recompute_scale_factor"):
-            m.recompute_scale_factor = None  # torch 1.11.0 compatibility
-
-    # Return model
-    if len(model) == 1:
-        return model[-1]
-
-    # Return detection ensemble
-    print(f"Ensemble created with {weights}\n")
-    for k in "names", "nc", "yaml":
-        setattr(model, k, getattr(model[0], k))
-    model.stride = model[torch.argmax(torch.tensor([m.stride.max() for m in model])).int()].stride  # max stride
-    assert all(model[0].nc == m.nc for m in model), f"Models have different class counts: {[m.nc for m in model]}"
-    return model

bplusplus/yolov5detect/models/hub/anchors.yaml

@@ -1,56 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-# Default anchors for COCO data
-
-# P5 -------------------------------------------------------------------------------------------------------------------
-# P5-640:
-anchors_p5_640:
-  - [10, 13, 16, 30, 33, 23] # P3/8
-  - [30, 61, 62, 45, 59, 119] # P4/16
-  - [116, 90, 156, 198, 373, 326] # P5/32
-
-# P6 -------------------------------------------------------------------------------------------------------------------
-# P6-640:  thr=0.25: 0.9964 BPR, 5.54 anchors past thr, n=12, img_size=640, metric_all=0.281/0.716-mean/best, past_thr=0.469-mean: 9,11,  21,19,  17,41,  43,32,  39,70,  86,64,  65,131,  134,130,  120,265,  282,180,  247,354,  512,387
-anchors_p6_640:
-  - [9, 11, 21, 19, 17, 41] # P3/8
-  - [43, 32, 39, 70, 86, 64] # P4/16
-  - [65, 131, 134, 130, 120, 265] # P5/32
-  - [282, 180, 247, 354, 512, 387] # P6/64
-
-# P6-1280:  thr=0.25: 0.9950 BPR, 5.55 anchors past thr, n=12, img_size=1280, metric_all=0.281/0.714-mean/best, past_thr=0.468-mean: 19,27,  44,40,  38,94,  96,68,  86,152,  180,137,  140,301,  303,264,  238,542,  436,615,  739,380,  925,792
-anchors_p6_1280:
-  - [19, 27, 44, 40, 38, 94] # P3/8
-  - [96, 68, 86, 152, 180, 137] # P4/16
-  - [140, 301, 303, 264, 238, 542] # P5/32
-  - [436, 615, 739, 380, 925, 792] # P6/64
-
-# P6-1920:  thr=0.25: 0.9950 BPR, 5.55 anchors past thr, n=12, img_size=1920, metric_all=0.281/0.714-mean/best, past_thr=0.468-mean: 28,41,  67,59,  57,141,  144,103,  129,227,  270,205,  209,452,  455,396,  358,812,  653,922,  1109,570,  1387,1187
-anchors_p6_1920:
-  - [28, 41, 67, 59, 57, 141] # P3/8
-  - [144, 103, 129, 227, 270, 205] # P4/16
-  - [209, 452, 455, 396, 358, 812] # P5/32
-  - [653, 922, 1109, 570, 1387, 1187] # P6/64
-
-# P7 -------------------------------------------------------------------------------------------------------------------
-# P7-640:  thr=0.25: 0.9962 BPR, 6.76 anchors past thr, n=15, img_size=640, metric_all=0.275/0.733-mean/best, past_thr=0.466-mean: 11,11,  13,30,  29,20,  30,46,  61,38,  39,92,  78,80,  146,66,  79,163,  149,150,  321,143,  157,303,  257,402,  359,290,  524,372
-anchors_p7_640:
-  - [11, 11, 13, 30, 29, 20] # P3/8
-  - [30, 46, 61, 38, 39, 92] # P4/16
-  - [78, 80, 146, 66, 79, 163] # P5/32
-  - [149, 150, 321, 143, 157, 303] # P6/64
-  - [257, 402, 359, 290, 524, 372] # P7/128
-
-# P7-1280:  thr=0.25: 0.9968 BPR, 6.71 anchors past thr, n=15, img_size=1280, metric_all=0.273/0.732-mean/best, past_thr=0.463-mean: 19,22,  54,36,  32,77,  70,83,  138,71,  75,173,  165,159,  148,334,  375,151,  334,317,  251,626,  499,474,  750,326,  534,814,  1079,818
-anchors_p7_1280:
-  - [19, 22, 54, 36, 32, 77] # P3/8
-  - [70, 83, 138, 71, 75, 173] # P4/16
-  - [165, 159, 148, 334, 375, 151] # P5/32
-  - [334, 317, 251, 626, 499, 474] # P6/64
-  - [750, 326, 534, 814, 1079, 818] # P7/128
-
-# P7-1920:  thr=0.25: 0.9968 BPR, 6.71 anchors past thr, n=15, img_size=1920, metric_all=0.273/0.732-mean/best, past_thr=0.463-mean: 29,34,  81,55,  47,115,  105,124,  207,107,  113,259,  247,238,  222,500,  563,227,  501,476,  376,939,  749,711,  1126,489,  801,1222,  1618,1227
-anchors_p7_1920:
-  - [29, 34, 81, 55, 47, 115] # P3/8
-  - [105, 124, 207, 107, 113, 259] # P4/16
-  - [247, 238, 222, 500, 563, 227] # P5/32
-  - [501, 476, 376, 939, 749, 711] # P6/64
-  - [1126, 489, 801, 1222, 1618, 1227] # P7/128

bplusplus/yolov5detect/models/hub/yolov3-spp.yaml

@@ -1,52 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors:
-  - [10, 13, 16, 30, 33, 23] # P3/8
-  - [30, 61, 62, 45, 59, 119] # P4/16
-  - [116, 90, 156, 198, 373, 326] # P5/32
-
-# darknet53 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [32, 3, 1]], # 0
-    [-1, 1, Conv, [64, 3, 2]], # 1-P1/2
-    [-1, 1, Bottleneck, [64]],
-    [-1, 1, Conv, [128, 3, 2]], # 3-P2/4
-    [-1, 2, Bottleneck, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 5-P3/8
-    [-1, 8, Bottleneck, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 7-P4/16
-    [-1, 8, Bottleneck, [512]],
-    [-1, 1, Conv, [1024, 3, 2]], # 9-P5/32
-    [-1, 4, Bottleneck, [1024]], # 10
-  ]
-
-# YOLOv3-SPP head
-head: [
-    [-1, 1, Bottleneck, [1024, False]],
-    [-1, 1, SPP, [512, [5, 9, 13]]],
-    [-1, 1, Conv, [1024, 3, 1]],
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, Conv, [1024, 3, 1]], # 15 (P5/32-large)
-
-    [-2, 1, Conv, [256, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 8], 1, Concat, [1]], # cat backbone P4
-    [-1, 1, Bottleneck, [512, False]],
-    [-1, 1, Bottleneck, [512, False]],
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, Conv, [512, 3, 1]], # 22 (P4/16-medium)
-
-    [-2, 1, Conv, [128, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P3
-    [-1, 1, Bottleneck, [256, False]],
-    [-1, 2, Bottleneck, [256, False]], # 27 (P3/8-small)
-
-    [[27, 22, 15], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
-  ]

bplusplus/yolov5detect/models/hub/yolov3-tiny.yaml

@@ -1,42 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors:
-  - [10, 14, 23, 27, 37, 58] # P4/16
-  - [81, 82, 135, 169, 344, 319] # P5/32
-
-# YOLOv3-tiny backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [16, 3, 1]], # 0
-    [-1, 1, nn.MaxPool2d, [2, 2, 0]], # 1-P1/2
-    [-1, 1, Conv, [32, 3, 1]],
-    [-1, 1, nn.MaxPool2d, [2, 2, 0]], # 3-P2/4
-    [-1, 1, Conv, [64, 3, 1]],
-    [-1, 1, nn.MaxPool2d, [2, 2, 0]], # 5-P3/8
-    [-1, 1, Conv, [128, 3, 1]],
-    [-1, 1, nn.MaxPool2d, [2, 2, 0]], # 7-P4/16
-    [-1, 1, Conv, [256, 3, 1]],
-    [-1, 1, nn.MaxPool2d, [2, 2, 0]], # 9-P5/32
-    [-1, 1, Conv, [512, 3, 1]],
-    [-1, 1, nn.ZeroPad2d, [[0, 1, 0, 1]]], # 11
-    [-1, 1, nn.MaxPool2d, [2, 1, 0]], # 12
-  ]
-
-# YOLOv3-tiny head
-head: [
-    [-1, 1, Conv, [1024, 3, 1]],
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, Conv, [512, 3, 1]], # 15 (P5/32-large)
-
-    [-2, 1, Conv, [128, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 8], 1, Concat, [1]], # cat backbone P4
-    [-1, 1, Conv, [256, 3, 1]], # 19 (P4/16-medium)
-
-    [[19, 15], 1, Detect, [nc, anchors]], # Detect(P4, P5)
-  ]

bplusplus/yolov5detect/models/hub/yolov3.yaml

@@ -1,52 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors:
-  - [10, 13, 16, 30, 33, 23] # P3/8
-  - [30, 61, 62, 45, 59, 119] # P4/16
-  - [116, 90, 156, 198, 373, 326] # P5/32
-
-# darknet53 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [32, 3, 1]], # 0
-    [-1, 1, Conv, [64, 3, 2]], # 1-P1/2
-    [-1, 1, Bottleneck, [64]],
-    [-1, 1, Conv, [128, 3, 2]], # 3-P2/4
-    [-1, 2, Bottleneck, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 5-P3/8
-    [-1, 8, Bottleneck, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 7-P4/16
-    [-1, 8, Bottleneck, [512]],
-    [-1, 1, Conv, [1024, 3, 2]], # 9-P5/32
-    [-1, 4, Bottleneck, [1024]], # 10
-  ]
-
-# YOLOv3 head
-head: [
-    [-1, 1, Bottleneck, [1024, False]],
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, Conv, [1024, 3, 1]],
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, Conv, [1024, 3, 1]], # 15 (P5/32-large)
-
-    [-2, 1, Conv, [256, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 8], 1, Concat, [1]], # cat backbone P4
-    [-1, 1, Bottleneck, [512, False]],
-    [-1, 1, Bottleneck, [512, False]],
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, Conv, [512, 3, 1]], # 22 (P4/16-medium)
-
-    [-2, 1, Conv, [128, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P3
-    [-1, 1, Bottleneck, [256, False]],
-    [-1, 2, Bottleneck, [256, False]], # 27 (P3/8-small)
-
-    [[27, 22, 15], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
-  ]

bplusplus/yolov5detect/models/hub/yolov5-bifpn.yaml

@@ -1,49 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors:
-  - [10, 13, 16, 30, 33, 23] # P3/8
-  - [30, 61, 62, 45, 59, 119] # P4/16
-  - [116, 90, 156, 198, 373, 326] # P5/32
-
-# YOLOv5 v6.0 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
-    [-1, 1, Conv, [128, 3, 2]], # 1-P2/4
-    [-1, 3, C3, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 3-P3/8
-    [-1, 6, C3, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 5-P4/16
-    [-1, 9, C3, [512]],
-    [-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
-    [-1, 3, C3, [1024]],
-    [-1, 1, SPPF, [1024, 5]], # 9
-  ]
-
-# YOLOv5 v6.0 BiFPN head
-head: [
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P4
-    [-1, 3, C3, [512, False]], # 13
-
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 4], 1, Concat, [1]], # cat backbone P3
-    [-1, 3, C3, [256, False]], # 17 (P3/8-small)
-
-    [-1, 1, Conv, [256, 3, 2]],
-    [[-1, 14, 6], 1, Concat, [1]], # cat P4 <--- BiFPN change
-    [-1, 3, C3, [512, False]], # 20 (P4/16-medium)
-
-    [-1, 1, Conv, [512, 3, 2]],
-    [[-1, 10], 1, Concat, [1]], # cat head P5
-    [-1, 3, C3, [1024, False]], # 23 (P5/32-large)
-
-    [[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
-  ]

bplusplus/yolov5detect/models/hub/yolov5-fpn.yaml

@@ -1,43 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors:
-  - [10, 13, 16, 30, 33, 23] # P3/8
-  - [30, 61, 62, 45, 59, 119] # P4/16
-  - [116, 90, 156, 198, 373, 326] # P5/32
-
-# YOLOv5 v6.0 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
-    [-1, 1, Conv, [128, 3, 2]], # 1-P2/4
-    [-1, 3, C3, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 3-P3/8
-    [-1, 6, C3, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 5-P4/16
-    [-1, 9, C3, [512]],
-    [-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
-    [-1, 3, C3, [1024]],
-    [-1, 1, SPPF, [1024, 5]], # 9
-  ]
-
-# YOLOv5 v6.0 FPN head
-head: [
-    [-1, 3, C3, [1024, False]], # 10 (P5/32-large)
-
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P4
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 3, C3, [512, False]], # 14 (P4/16-medium)
-
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 4], 1, Concat, [1]], # cat backbone P3
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 3, C3, [256, False]], # 18 (P3/8-small)
-
-    [[18, 14, 10], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
-  ]

bplusplus/yolov5detect/models/hub/yolov5-p2.yaml

@@ -1,55 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors: 3 # AutoAnchor evolves 3 anchors per P output layer
-
-# YOLOv5 v6.0 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
-    [-1, 1, Conv, [128, 3, 2]], # 1-P2/4
-    [-1, 3, C3, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 3-P3/8
-    [-1, 6, C3, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 5-P4/16
-    [-1, 9, C3, [512]],
-    [-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
-    [-1, 3, C3, [1024]],
-    [-1, 1, SPPF, [1024, 5]], # 9
-  ]
-
-# YOLOv5 v6.0 head with (P2, P3, P4, P5) outputs
-head: [
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P4
-    [-1, 3, C3, [512, False]], # 13
-
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 4], 1, Concat, [1]], # cat backbone P3
-    [-1, 3, C3, [256, False]], # 17 (P3/8-small)
-
-    [-1, 1, Conv, [128, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 2], 1, Concat, [1]], # cat backbone P2
-    [-1, 1, C3, [128, False]], # 21 (P2/4-xsmall)
-
-    [-1, 1, Conv, [128, 3, 2]],
-    [[-1, 18], 1, Concat, [1]], # cat head P3
-    [-1, 3, C3, [256, False]], # 24 (P3/8-small)
-
-    [-1, 1, Conv, [256, 3, 2]],
-    [[-1, 14], 1, Concat, [1]], # cat head P4
-    [-1, 3, C3, [512, False]], # 27 (P4/16-medium)
-
-    [-1, 1, Conv, [512, 3, 2]],
-    [[-1, 10], 1, Concat, [1]], # cat head P5
-    [-1, 3, C3, [1024, False]], # 30 (P5/32-large)
-
-    [[21, 24, 27, 30], 1, Detect, [nc, anchors]], # Detect(P2, P3, P4, P5)
-  ]

bplusplus/yolov5detect/models/hub/yolov5-p34.yaml

@@ -1,42 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 0.33 # model depth multiple
-width_multiple: 0.50 # layer channel multiple
-anchors: 3 # AutoAnchor evolves 3 anchors per P output layer
-
-# YOLOv5 v6.0 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
-    [-1, 1, Conv, [128, 3, 2]], # 1-P2/4
-    [-1, 3, C3, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 3-P3/8
-    [-1, 6, C3, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 5-P4/16
-    [-1, 9, C3, [512]],
-    [-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
-    [-1, 3, C3, [1024]],
-    [-1, 1, SPPF, [1024, 5]], # 9
-  ]
-
-# YOLOv5 v6.0 head with (P3, P4) outputs
-head: [
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P4
-    [-1, 3, C3, [512, False]], # 13
-
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 4], 1, Concat, [1]], # cat backbone P3
-    [-1, 3, C3, [256, False]], # 17 (P3/8-small)
-
-    [-1, 1, Conv, [256, 3, 2]],
-    [[-1, 14], 1, Concat, [1]], # cat head P4
-    [-1, 3, C3, [512, False]], # 20 (P4/16-medium)
-
-    [[17, 20], 1, Detect, [nc, anchors]], # Detect(P3, P4)
-  ]

bplusplus/yolov5detect/models/hub/yolov5-p6.yaml

@@ -1,57 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors: 3 # AutoAnchor evolves 3 anchors per P output layer
-
-# YOLOv5 v6.0 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
-    [-1, 1, Conv, [128, 3, 2]], # 1-P2/4
-    [-1, 3, C3, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 3-P3/8
-    [-1, 6, C3, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 5-P4/16
-    [-1, 9, C3, [512]],
-    [-1, 1, Conv, [768, 3, 2]], # 7-P5/32
-    [-1, 3, C3, [768]],
-    [-1, 1, Conv, [1024, 3, 2]], # 9-P6/64
-    [-1, 3, C3, [1024]],
-    [-1, 1, SPPF, [1024, 5]], # 11
-  ]
-
-# YOLOv5 v6.0 head with (P3, P4, P5, P6) outputs
-head: [
-    [-1, 1, Conv, [768, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 8], 1, Concat, [1]], # cat backbone P5
-    [-1, 3, C3, [768, False]], # 15
-
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P4
-    [-1, 3, C3, [512, False]], # 19
-
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 4], 1, Concat, [1]], # cat backbone P3
-    [-1, 3, C3, [256, False]], # 23 (P3/8-small)
-
-    [-1, 1, Conv, [256, 3, 2]],
-    [[-1, 20], 1, Concat, [1]], # cat head P4
-    [-1, 3, C3, [512, False]], # 26 (P4/16-medium)
-
-    [-1, 1, Conv, [512, 3, 2]],
-    [[-1, 16], 1, Concat, [1]], # cat head P5
-    [-1, 3, C3, [768, False]], # 29 (P5/32-large)
-
-    [-1, 1, Conv, [768, 3, 2]],
-    [[-1, 12], 1, Concat, [1]], # cat head P6
-    [-1, 3, C3, [1024, False]], # 32 (P6/64-xlarge)
-
-    [[23, 26, 29, 32], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5, P6)
-  ]

bplusplus/yolov5detect/models/hub/yolov5-p7.yaml

@@ -1,68 +0,0 @@
-# Ultralytics YOLOv5 🚀, AGPL-3.0 license
-
-# Parameters
-nc: 80 # number of classes
-depth_multiple: 1.0 # model depth multiple
-width_multiple: 1.0 # layer channel multiple
-anchors: 3 # AutoAnchor evolves 3 anchors per P output layer
-
-# YOLOv5 v6.0 backbone
-backbone:
-  # [from, number, module, args]
-  [
-    [-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
-    [-1, 1, Conv, [128, 3, 2]], # 1-P2/4
-    [-1, 3, C3, [128]],
-    [-1, 1, Conv, [256, 3, 2]], # 3-P3/8
-    [-1, 6, C3, [256]],
-    [-1, 1, Conv, [512, 3, 2]], # 5-P4/16
-    [-1, 9, C3, [512]],
-    [-1, 1, Conv, [768, 3, 2]], # 7-P5/32
-    [-1, 3, C3, [768]],
-    [-1, 1, Conv, [1024, 3, 2]], # 9-P6/64
-    [-1, 3, C3, [1024]],
-    [-1, 1, Conv, [1280, 3, 2]], # 11-P7/128
-    [-1, 3, C3, [1280]],
-    [-1, 1, SPPF, [1280, 5]], # 13
-  ]
-
-# YOLOv5 v6.0 head with (P3, P4, P5, P6, P7) outputs
-head: [
-    [-1, 1, Conv, [1024, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 10], 1, Concat, [1]], # cat backbone P6
-    [-1, 3, C3, [1024, False]], # 17
-
-    [-1, 1, Conv, [768, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 8], 1, Concat, [1]], # cat backbone P5
-    [-1, 3, C3, [768, False]], # 21
-
-    [-1, 1, Conv, [512, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 6], 1, Concat, [1]], # cat backbone P4
-    [-1, 3, C3, [512, False]], # 25
-
-    [-1, 1, Conv, [256, 1, 1]],
-    [-1, 1, nn.Upsample, [None, 2, "nearest"]],
-    [[-1, 4], 1, Concat, [1]], # cat backbone P3
-    [-1, 3, C3, [256, False]], # 29 (P3/8-small)
-
-    [-1, 1, Conv, [256, 3, 2]],
-    [[-1, 26], 1, Concat, [1]], # cat head P4
-    [-1, 3, C3, [512, False]], # 32 (P4/16-medium)
-
-    [-1, 1, Conv, [512, 3, 2]],
-    [[-1, 22], 1, Concat, [1]], # cat head P5
-    [-1, 3, C3, [768, False]], # 35 (P5/32-large)
-
-    [-1, 1, Conv, [768, 3, 2]],
-    [[-1, 18], 1, Concat, [1]], # cat head P6
-    [-1, 3, C3, [1024, False]], # 38 (P6/64-xlarge)
-
-    [-1, 1, Conv, [1024, 3, 2]],
-    [[-1, 14], 1, Concat, [1]], # cat head P7
-    [-1, 3, C3, [1280, False]], # 41 (P7/128-xxlarge)
-
-    [[29, 32, 35, 38, 41], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5, P6, P7)
-  ]