diff --git a/logo_matching.py b/logo_matching.py index 61ba7c8..6d4ff43 100644 --- a/logo_matching.py +++ b/logo_matching.py @@ -23,7 +23,7 @@ def check_domain_brand_inconsistency(logo_boxes, with open(domain_map_path, 'rb') as handle: domain_map = pickle.load(handle) with open(os.path.join(os.path.dirname(__file__), 'configs.yaml')) as file: - configs = yaml.load(file, Loader = yaml.FullLoader) + configs = yaml.load(file, Loader=yaml.FullLoader) zip_path = os.path.join(os.path.dirname(__file__), configs['SIAMESE_MODEL']['TARGETLIST_PATH']) # 获取解压后的目录路径 targetlist_path = os.path.join(os.path.dirname(zip_path), os.path.basename(zip_path).split('.zip')[0])