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

correct variable names in base_yolo_neck (#162)

pull/249/head
Range King 2022-10-14 10:01:27 +08:00 committed by Haian Huang(深度眸)
parent db73593a15
commit 9ef467cd6c
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mmyolo/models/necks/base_yolo_neck.py
View File

@ -128,11 +128,11 @@ class BaseYOLONeck(BaseModule, metaclass=ABCMeta):
# top-down path
inner_outs = [reduce_outs[-1]]
for idx in range(len(self.in_channels) - 1, 0, -1):
feat_heigh = inner_outs[0]
feat_high = inner_outs[0]
feat_low = reduce_outs[idx - 1]
upsample_feat = self.upsample_layers[len(self.in_channels) - 1 -
idx](
feat_heigh)
feat_high)
inner_out = self.top_down_layers[len(self.in_channels) - 1 - idx](
torch.cat([upsample_feat, feat_low], 1))
@ -142,10 +142,10 @@ class BaseYOLONeck(BaseModule, metaclass=ABCMeta):
outs = [inner_outs[0]]
for idx in range(len(self.in_channels) - 1):
feat_low = outs[-1]
feat_height = inner_outs[idx + 1]
feat_high = inner_outs[idx + 1]
downsample_feat = self.downsample_layers[idx](feat_low)
out = self.bottom_up_layers[idx](
torch.cat([downsample_feat, feat_height], 1))
torch.cat([downsample_feat, feat_high], 1))
outs.append(out)
# out_layers