From d40cad5898d12a083e9e06f4094f58150b2b54f8 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 1 Oct 2021 20:23:10 -0700
Subject: [PATCH] Update yolo.py

---
 models/yolo.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/models/yolo.py b/models/yolo.py
index 3c7b9ecda9dc..796554211272 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -172,13 +172,13 @@ def _clip_augmented(self, y):
         nl = self.model[-1].nl  # number of detection layers (P3-P5)
         n = sum(4 ** x for x in range(nl))
 
-        # small
+        # large
         i = y[0].shape[1] // n
-        y[0] = y[0][:, i:]
+        y[0] = y[0][:, :-i]
 
-        # large
+        # small
         i = (y[-1].shape[1] // n) * 4 ** (nl - 1)
-        y[-1] = y[-1][:, :-i]
+        y[-1] = y[-1][:, i:]
 
         return y