mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Bug] Support RTMDet P6 model (#480)

pull/482/head
Haian Huang(深度眸) 2023-01-18 15:02:39 +08:00 committed by GitHub
parent da62ae7283
commit 6f478e1ae0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmyolo/models/necks/cspnext_pafpn.py
View File

@ -90,7 +90,7 @@ class CSPNeXtPAFPN(BaseYOLONeck):
Returns:
nn.Module: The reduce layer.
"""
if idx == 2:
if idx == len(self.in_channels) - 1:
layer = self.conv(
self.in_channels[idx],
self.in_channels[idx - 1],