From fc341c91e5c1efff3ff63e8111d2a9f02599251e Mon Sep 17 00:00:00 2001 From: "Anders U. Waldeland" <32647204+waldeland@users.noreply.github.com> Date: Tue, 5 Mar 2019 16:03:33 +0100 Subject: [PATCH] Make the ordering of prior_boxes deterministic In the current implementation the ordering of the prior_boxes-list is random. This makes it impossible to load pretrained model weights (without the model definition). --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index 9deaedf..080a1f0 100644 --- a/model.py +++ b/model.py @@ -395,7 +395,7 @@ def create_prior_boxes(self): 'conv10_2': [1., 2., 0.5], 'conv11_2': [1., 2., 0.5]} - fmaps = list(fmap_dims.keys()) + fmaps = sorted(list(fmap_dims.keys())) prior_boxes = []