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

removed xy/wh loss reporting

This commit is contained in:
Glenn Jocher 2019-08-24 21:35:56 +02:00
parent ca38c9050f
commit 1064c37600
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -268,11 +268,11 @@ def train():
'%g/%g' % (epoch, epochs - 1), '%.3gG' % mem, *mloss, len(targets), img_size) '%g/%g' % (epoch, epochs - 1), '%.3gG' % mem, *mloss, len(targets), img_size)
pbar.set_description(s) # end batch ----------------------------------------------------------------------- pbar.set_description(s) # end batch -----------------------------------------------------------------------
final_epoch = epoch + 1 == epochs
if opt.prebias: if opt.prebias:
print_model_biases(model) print_model_biases(model)
else: else:
# Calculate mAP (always test final epoch, skip first 10 if opt.nosave) # Calculate mAP (always test final epoch, skip first 10 if opt.nosave)
final_epoch = epoch + 1 == epochs
if not (opt.notest or (opt.nosave and epoch < 10)) or final_epoch: if not (opt.notest or (opt.nosave and epoch < 10)) or final_epoch:
with torch.no_grad(): with torch.no_grad():
results, maps = test.test(cfg, results, maps = test.test(cfg,

1
utils/utils.py
View File

@ -555,6 +555,7 @@ def get_yolo_layers(model):
def print_model_biases(model): def print_model_biases(model):
# prints the bias neurons preceding each yolo layer # prints the bias neurons preceding each yolo layer
print('\nModel Output-Bias Summary::')
for l in model.yolo_layers: # print pretrained biases for l in model.yolo_layers: # print pretrained biases
b = model.module_list[l - 1][0].bias.view(3, -1) # bias 3x85 b = model.module_list[l - 1][0].bias.view(3, -1) # bias 3x85
print('regression: %.2f+/-%.2f, ' % (b[:, :4].mean(), b[:, :4].std()), print('regression: %.2f+/-%.2f, ' % (b[:, :4].mean(), b[:, :4].std()),