From 6fe67595cb856fe9b8d8c745c241aec3f772731b Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 15 May 2020 11:32:23 -0700
Subject: [PATCH] add stride order reversal for c53*.cfg

---
 models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models.py b/models.py
index ebe151b6..a85bf987 100755
--- a/models.py
+++ b/models.py
@@ -93,7 +93,7 @@ def create_modules(module_defs, img_size, cfg):
         elif mdef['type'] == 'yolo':
             yolo_index += 1
             stride = [32, 16, 8]  # P5, P4, P3 strides
-            if 'panet' in cfg or 'yolov4' in cfg:  # stride order reversed
+            if 'panet' in cfg or 'yolov4' or 'cd53' in cfg:  # stride order reversed
                 stride = list(reversed(stride))
             layers = mdef['from'] if 'from' in mdef else []
             modules = YOLOLayer(anchors=mdef['anchors'][mdef['mask']],  # anchor list