Sfoglia il codice sorgente

Set logger level by argument

Kentaro Wada 6 anni fa
parent
commit
f41f332ff1
2 ha cambiato i file con 10 aggiunte e 0 eliminazioni
  1. 1 0
      labelme/config/default_config.yaml
  2. 9 0
      labelme/main.py

+ 1 - 0
labelme/config/default_config.yaml

@@ -2,6 +2,7 @@ auto_save: false
 display_label_popup: true
 store_data: true
 keep_prev: false
+logger_level: info
 
 flags: null
 labels: null

+ 9 - 0
labelme/main.py

@@ -1,5 +1,6 @@
 import argparse
 import codecs
+import logging
 import os
 import sys
 
@@ -28,6 +29,12 @@ def _main():
     parser.add_argument(
         '--reset-config', action='store_true', help='reset qt config'
     )
+    parser.add_argument(
+        '--logger-level',
+        default='info',
+        choices=['debug', 'info', 'warning', 'fatal', 'error'],
+        help='logger level',
+    )
     parser.add_argument('filename', nargs='?', help='image or label filename')
     parser.add_argument(
         '--output',
@@ -100,6 +107,8 @@ def _main():
         print('{0} {1}'.format(__appname__, __version__))
         sys.exit(0)
 
+    logger.setLevel(getattr(logging, args.logger_level.upper()))
+
     if hasattr(args, 'flags'):
         if os.path.isfile(args.flags):
             with codecs.open(args.flags, 'r', encoding='utf-8') as f: