configfile

bmd1905 · Dec 9, 2020 · 6816da2 · 6816da2
1 parent baa7f46
commit 6816da2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vietocr/tool/config.py b/vietocr/tool/config.py
@@ -5,7 +5,7 @@
         'vgg_transformer':'vgg-transformer.yml',
         'resnet_transformer':'resnet_transformer.yml',
         'resnet_fpn_transformer':'resnet_fpn_transformer.yml',
-        'vgg_seq2seq':'vgg_seq2seq.yml',
+        'vgg_seq2seq':'vgg-seq2seq.yml',
         'vgg_convseq2seq':'vgg_convseq2seq.yml',
         'vgg_decoderseq2seq':'vgg_decoderseq2seq.yml',
         'base':'base.yml',

diff --git a/vietocr/tool/utils.py b/vietocr/tool/utils.py
@@ -18,7 +18,7 @@ def download_config(id):
     url = 'https://raw.githubusercontent.com/pbcquoc/vietocr/master/config/{}'.format(id)
     r = requests.get(url)
     config = yaml.safe_load(r.text)
-
+    print(config)
     return config
 
 def compute_accuracy(ground_truth, predictions, mode='full_sequence'):