Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2020-02-04 21:22:20 -08:00
parent 888cad1e31
commit e185719bd7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -268,7 +268,7 @@ class Darknet(nn.Module):
x = [torch.cat(x, 0) for x in zip(*output)] x = [torch.cat(x, 0) for x in zip(*output)]
return x[0], torch.cat(x[1:3], 1) # scores, boxes: 3780x80, 3780x4 return x[0], torch.cat(x[1:3], 1) # scores, boxes: 3780x80, 3780x4
else: else:
io, p = list(zip(*output)) # inference output, training output io, p = zip(*output) # inference output, training output
return torch.cat(io, 1), p return torch.cat(io, 1), p
def fuse(self): def fuse(self):