From fc1bc51b032628cf3fc3c0b346ab44f181877a26 Mon Sep 17 00:00:00 2001 From: Sacha Date: Fri, 9 Oct 2020 14:17:03 +1000 Subject: [PATCH] Fix efficientnet-b4 Requires width factor of 1.5 to be correct size for input --- backbone/efficientnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backbone/efficientnet.py b/backbone/efficientnet.py index bf79dcd..4096539 100644 --- a/backbone/efficientnet.py +++ b/backbone/efficientnet.py @@ -387,7 +387,7 @@ def efficientnet(input_size, embedding_size=512, version='b1', **kwargs): elif version == "b4": in_size = (190, 190) depth_factor = 1.8 - width_factor = 1.4 + width_factor = 1.5 dropout_rate = 0.4 elif version == "b5": in_size = (228, 228) @@ -455,4 +455,4 @@ def efficientnet(input_size, embedding_size=512, version='b1', **kwargs): embedding_size=embedding_size, **kwargs) - return net \ No newline at end of file + return net