From 20891926c9f721a338ff6183514819f2fa8077f6 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 15 May 2020 14:40:14 -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 a85bf987..0c0d6fa0 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' or 'cd53' in cfg:  # stride order reversed
+            if any(x in cfg for x in ['panet', 'yolov4', 'cd53']):  # stride order reversed
                 stride = list(reversed(stride))
             layers = mdef['from'] if 'from' in mdef else []
             modules = YOLOLayer(anchors=mdef['anchors'][mdef['mask']],  # anchor list