From 5dcb0be6795ff9fe9c30dd84d6215fb5ef7874df Mon Sep 17 00:00:00 2001 From: liuhui29 Date: Fri, 18 Sep 2020 15:13:04 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8logger=E6=89=93=E5=8D=B0?= =?UTF-8?q?=E6=97=A5=E5=BF=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools/create_label_list.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/tools/create_label_list.py b/tools/create_label_list.py index e5fd266d0..1e2ae9ecf 100644 --- a/tools/create_label_list.py +++ b/tools/create_label_list.py @@ -27,6 +27,11 @@ import xml.etree.ElementTree as ET from tqdm import tqdm import re +import logging +FORMAT = '%(asctime)s-%(levelname)s: %(message)s' +logging.basicConfig(level=logging.INFO, format=FORMAT) +logger = logging.getLogger(__name__) + def main(): """ @@ -49,13 +54,16 @@ def main(): default='output/label_list.txt') args = parser.parse_args() + args.dataset_type = args.dataset_type.lower() + assert args.dataset_type in ['coco', 'voc'] - if args.dataset_type == 'coco' or args.dataset_type == 'COCO': + if args.dataset_type == 'coco': assert args.dataset_type and args.json_path try: assert os.path.exists(args.json_path) except AssertionError as e: - print('The json file: {} does not exist!'.format(args.json_path)) + logger.error('The json file: {} does not exist!'.format( + args.json_path)) os._exit(0) anno = json.load(open(args.json_path)) @@ -66,12 +74,12 @@ def main(): f.write(cat + '\n') print('lable_list file: {} create done!'.format(args.outfile_path)) - if args.dataset_type == 'voc' or args.dataset_type == 'VOC': + if args.dataset_type == 'voc': assert args.dataset_type and args.xml_dir try: assert os.path.exists(args.xml_dir) except AssertionError as e: - print('The xml_dir: {} does not exist!'.format(args.xml_dir)) + logger.error('The xml_dir: {} does not exist!'.format(args.xml_dir)) os._exit(0) categories_list = set() @@ -89,7 +97,8 @@ def main(): with open(args.outfile_path, 'w') as f: for cat in categories_list: f.write(cat + '\n') - print('lable_list file: {} create done!'.format(args.outfile_path)) + logger.info('lable_list file: {} create done!'.format( + args.outfile_path)) if __name__ == '__main__': -- GitLab