From 3ee457cd3da386745c8b354eacb1b1ac01d42f48 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 24 Aug 2019 21:50:33 +0200 Subject: [PATCH] updates --- utils/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index 94858c30..2eaa61b1 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -555,12 +555,12 @@ def get_yolo_layers(model): def print_model_biases(model): # prints the bias neurons preceding each yolo layer - print('\nModel output-bias Summary:') + print('\nModel Bias Summary (per output layer):') 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())) + print('regression: %5.2f+/-%-5.2f ' % (b[:, :4].mean(), b[:, :4].std()), + 'objectness: %5.2f+/-%-5.2f ' % (b[:, 4].mean(), b[:, 4].std()), + 'classification: %5.2f+/-%-5.2f' % (b[:, 5:].mean(), b[:, 5:].std())) def strip_optimizer(f='weights/last.pt'): # from utils.utils import *; strip_optimizer()