From 57ecc1c4dbf9b72800fcc9f28989dfc20425c681 Mon Sep 17 00:00:00 2001
From: xiatao <35915584+dreamer121121@users.noreply.github.com>
Date: Sun, 28 Aug 2022 14:33:55 +0800
Subject: [PATCH] Update yolo.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

解决 SPD_yolov5, parse_model错误
---
 models/yolo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/yolo.py b/models/yolo.py
index 52f643d..6afc2cf 100644
--- a/models/yolo.py
+++ b/models/yolo.py
@@ -314,7 +314,7 @@ def parse_model(d, ch):  # model_dict, input_channels(3)
         RepResX, RepResXCSPA, RepResXCSPB, RepResXCSPC, Ghost, GhostCSPA, GhostCSPB, GhostCSPC,
         SwinTransformerBlock, STCSPA, STCSPB, STCSPC,
         SwinTransformer2Block, ST2CSPA, ST2CSPB, ST2CSPC,
-        conv_bn_relu_maxpool, Shuffle_Block, RepVGGBlock, CBH, LC_Block, Dense, DWConvblock, ShuffleNetV2x, DWConvblockX, space_to_depth,
+        conv_bn_relu_maxpool, Shuffle_Block, RepVGGBlock, CBH, LC_Block, Dense, DWConvblock, ShuffleNetV2x, DWConvblockX,
         SEAM, RFEM, C3RFEM, ConvMixer, MultiSEAM):
             c1, c2 = ch[f], args[0]
             if c2 != no:  # if not output