From d1398ec9521d1d0a95f564cf5b25a35a8e5beef5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 7 Oct 2019 11:31:22 +0200 Subject: [PATCH] updates --- utils/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index da81dbe..e990976 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -560,9 +560,11 @@ def print_model_biases(model): multi_gpu = type(model) in (nn.parallel.DataParallel, nn.parallel.DistributedDataParallel) for l in model.yolo_layers: # print pretrained biases if multi_gpu: - b = model.module.module_list[l - 1][0].bias.view(3, -1) # bias 3x85 + na = model.module.module_list[l].na # number of anchors + b = model.module.module_list[l - 1][0].bias.view(na, -1) # bias 3x85 else: - b = model.module_list[l - 1][0].bias.view(3, -1) # bias 3x85 + na = model.module_list[l].na + b = model.module_list[l - 1][0].bias.view(na, -1) # bias 3x85 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()))