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

augment update

This commit is contained in:
Glenn Jocher 2020-04-07 17:35:35 -07:00
parent 6c5ecaf805
commit 933e05cb44
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models.py
View File

@ -252,9 +252,9 @@ class Darknet(nn.Module):
# for i, yi in enumerate(y): # coco small, medium, large = < 32**2 < 96**2 < # for i, yi in enumerate(y): # coco small, medium, large = < 32**2 < 96**2 <
# area = yi[..., 2:4].prod(2)[:, :, None] # area = yi[..., 2:4].prod(2)[:, :, None]
# if i == 1: # if i == 1:
# yi = yi * (area < 96. ** 2).float() # yi *= (area < 96. ** 2).float()
# elif i == 2: # elif i == 2:
# yi = yi * (area > 32. ** 2).float() # yi *= (area > 32. ** 2).float()
# y[i] = yi # y[i] = yi
y = torch.cat(y, 1) y = torch.cat(y, 1)