From 2d92c076225efbe26195a49a5e79b9f0f65e77c0 Mon Sep 17 00:00:00 2001 From: breezedeus Date: Tue, 5 May 2020 11:57:50 +0800 Subject: [PATCH] finetune the arch of densenet --- cnocr/symbols/densenet.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/cnocr/symbols/densenet.py b/cnocr/symbols/densenet.py index 6eddfc4..fa575fd 100644 --- a/cnocr/symbols/densenet.py +++ b/cnocr/symbols/densenet.py @@ -217,13 +217,13 @@ def _make_final_stage_net(stage_index, out_channels): with features.name_scope(): features.add(nn.BatchNorm()) features.add(nn.Activation('relu')) - features.add(nn.Conv2D(out_channels // 4, kernel_size=1, use_bias=False)) - features.add(nn.BatchNorm()) - features.add(nn.Activation('relu')) - features.add( - nn.Conv2D(out_channels, kernel_size=(2, 1), strides=(2, 1), use_bias=False) - ) - features.add(nn.BatchNorm()) - features.add(nn.Activation('relu')) - # features.add(nn.MaxPool2D(pool_size=(2, 1), strides=(2, 1))) + # features.add(nn.Conv2D(out_channels // 4, kernel_size=1, use_bias=False)) + # features.add(nn.BatchNorm()) + # features.add(nn.Activation('relu')) + # features.add( + # nn.Conv2D(out_channels, kernel_size=(2, 1), strides=(2, 1), use_bias=False) + # ) + # features.add(nn.BatchNorm()) + # features.add(nn.Activation('relu')) + features.add(nn.MaxPool2D(pool_size=(2, 1), strides=(2, 1))) return features -- GitLab