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

Focal Loss bias initialization

This commit is contained in:
Glenn Jocher 2019-08-19 01:27:41 +02:00
parent 96fd7141a2
commit 98a24c0a2f
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
models.py
View File

@ -74,6 +74,12 @@ def create_modules(module_defs, img_size):
nc=int(mdef['classes']), # number of classes nc=int(mdef['classes']), # number of classes
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)
bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85
bias[:, 4:] -= - 5
module_list[-1][0].bias = torch.nn.Parameter(bias.view(-1))
else: else:
print('Warning: Unrecognized Layer Type: ' + mdef['type']) print('Warning: Unrecognized Layer Type: ' + mdef['type'])