Squash Reverts

This reverts commit 92d9cc2.

Revert "Updated yolox"

This reverts commit 5f7a441.

Revert "Updated UOLOx"

This reverts commit 6d9eeac.

Revert "In chanels for UoloX"

This reverts commit 41fd727.

Revert "In channels UoloX2"

This reverts commit 3cb9924.

Revert "In channels UoloX3"

This reverts commit 1046e08.

models/common.py

@@ -858,8 +858,8 @@ def __init__(self, in_channels):
         # self.cv22 = Conv(48, 16, k=3)
 
         self.upsample = nn.Upsample(scale_factor=2, mode='nearest')
-        self.cv2 = Conv(256, 80, k=3)
-        self.cv3 = Conv(160, 16, k=3)
+        self.cv2 = Conv(192, 48, k=3)
+        self.cv3 = Conv(96, 16, k=3)
         self.cv4 = Conv(16, 1, act=False)
         self.relu = nn.ReLU()
         self.dropout_normal = nn.Dropout(0.5)
@@ -874,7 +874,6 @@ def forward(self, x, skipped_input):
         x = self.cv2(x)
         x = self.upsample(x)
         # x2 = self.cv22(skipped_input[1])
-        # print(skipped_input[1].shape)
         x = torch.cat((x, skipped_input[1]), 1)  # Skip connection
 
         x = self.cv3(x)

models/yolov5x.yaml

@@ -44,5 +44,5 @@ head:
    [[-1, 10], 1, Concat, [1]],  # cat head P5
    [-1, 3, C3, [1024, False]],  # 23 (P5/32-large)
 
-   [[17, 20, 23, 2, 0], 1, SemanticSegment, [nc, anchors]],  # Detect(P3, P4, P5)
+   [[17, 20, 23], 1, Detect, [nc, anchors]],  # Detect(P3, P4, P5)
   ]