revise format

This commit is contained in:
rujiao.lrj
2022-11-17 15:36:07 +08:00
parent 4cf627ec56
commit 3d781dfcde
2 changed files with 1 additions and 2 deletions

View File

@@ -635,7 +635,7 @@ class DLAUp(nn.Module):
"ida_{}".format(i),
IDAUp(3, channels[j], in_channels[j:], scales[j:] // scales[j]),
)
scales[j + 1 :] = scales[j]
scales[j + 1:] = scales[j]
in_channels[j + 1 :] = [channels[j] for _ in channels[j + 1 :]]
def forward(self, layers):

View File

@@ -212,7 +212,6 @@ def gbox_post_process(gbox, c, s, h, w):
def nms(dets, thresh):
if len(dets) < 2:
return dets
scores = dets[:, 8]
index_keep = []
keep = []
for i in range(len(dets)):