From 7e9b53abe35f9458fefcf24f227ea7c262ba2aa1 Mon Sep 17 00:00:00 2001 From: Alireza Kashani Date: Thu, 4 Jan 2024 15:39:54 +0100 Subject: [PATCH] Update layout_analysis.py Dictionary comprehension is typically faster and more Pythonic than using a for loop. --- Applications/DocXChain/modules/layout_analysis.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/Applications/DocXChain/modules/layout_analysis.py b/Applications/DocXChain/modules/layout_analysis.py index 6702517..5d4a40d 100644 --- a/Applications/DocXChain/modules/layout_analysis.py +++ b/Applications/DocXChain/modules/layout_analysis.py @@ -36,9 +36,7 @@ def __init__(self, configs): # load map information map_info = json.load(open(BASE_DIR + '/../../../DocumentUnderstanding/DocXLayout/map_info.json')) - category_map = {} - for cate, idx in map_info["huntie"]["primary_map"].items(): - category_map[idx] = cate + category_map = {idx: cate for cate, idx in map_info["huntie"]["primary_map"].items()} self.category_map = category_map @@ -88,4 +86,4 @@ def release(self): if self.layout_analyser is not None: del self.layout_analyser - return \ No newline at end of file + return