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

updates

This commit is contained in:
Glenn Jocher 2019-08-19 17:07:16 +02:00
parent e1f724b1a3
commit d94ccd105d
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
models.py
View File

@ -75,11 +75,18 @@ def create_modules(module_defs, img_size):
img_size=img_size, # (416, 416) img_size=img_size, # (416, 416)
yolo_index=yolo_index) # 0, 1 or 2 yolo_index=yolo_index) # 0, 1 or 2
# Initialize preceding Conv2d() detection bias to -5 (https://arxiv.org/pdf/1708.02002.pdf section 3.3) # Initialize preceding Conv2d() bias (https://arxiv.org/pdf/1708.02002.pdf section 3.3)
bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85 bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85
bias[:, 4:] -= 5 bias[:, 4] -= 3.0 # obj
bias[:, 5:] -= 0.3 # cls
module_list[-1][0].bias = torch.nn.Parameter(bias.view(-1)) module_list[-1][0].bias = torch.nn.Parameter(bias.view(-1))
# for l in model.yolo_layers: # print pretrained biases
# b = model.module_list[l - 1][0].bias.view(3, -1) # bias 3x85
# print('regression: %.2f+/-%.2f, ' % (b[:, :4].mean(), b[:, :4].std()),
# 'objectness: %.2f+/-%.2f, ' % (b[:, 4].mean(), b[:, 4].std()),
# 'classification: %.2f+/-%.2f' % (b[:, 5:].mean(), b[:, 5:].std()))
else: else:
print('Warning: Unrecognized Layer Type: ' + mdef['type']) print('Warning: Unrecognized Layer Type: ' + mdef['type'])