feat: add runtime exclusion list
Browse files
font_ds_generate_script.py
CHANGED
|
@@ -32,7 +32,7 @@ dataset_path = "./dataset/font_img"
|
|
| 32 |
os.makedirs(dataset_path, exist_ok=True)
|
| 33 |
|
| 34 |
unqualified_log_file_name = f"unqualified_font_{time.time()}.txt"
|
| 35 |
-
|
| 36 |
|
| 37 |
fonts, exclusion_rule = load_fonts()
|
| 38 |
corpus_manager = CorpusGeneratorManager()
|
|
@@ -52,6 +52,9 @@ def generate_dataset(dataset_type: str, cnt: int):
|
|
| 52 |
if exclusion_rule(font):
|
| 53 |
print(f"Excluded font: {font.path}")
|
| 54 |
return
|
|
|
|
|
|
|
|
|
|
| 55 |
|
| 56 |
while True:
|
| 57 |
try:
|
|
@@ -77,6 +80,7 @@ def generate_dataset(dataset_type: str, cnt: int):
|
|
| 77 |
return
|
| 78 |
except UnqualifiedFontException as e:
|
| 79 |
print(f"SKIPPING Unqualified font: {e.font.path}")
|
|
|
|
| 80 |
with open(unqualified_log_file_name, "a+") as f:
|
| 81 |
f.write(f"{e.font.path}\n")
|
| 82 |
return
|
|
|
|
| 32 |
os.makedirs(dataset_path, exist_ok=True)
|
| 33 |
|
| 34 |
unqualified_log_file_name = f"unqualified_font_{time.time()}.txt"
|
| 35 |
+
runtime_exclusion_list = []
|
| 36 |
|
| 37 |
fonts, exclusion_rule = load_fonts()
|
| 38 |
corpus_manager = CorpusGeneratorManager()
|
|
|
|
| 52 |
if exclusion_rule(font):
|
| 53 |
print(f"Excluded font: {font.path}")
|
| 54 |
return
|
| 55 |
+
if font.path in runtime_exclusion_list:
|
| 56 |
+
print(f"Excluded font: {font.path}")
|
| 57 |
+
return
|
| 58 |
|
| 59 |
while True:
|
| 60 |
try:
|
|
|
|
| 80 |
return
|
| 81 |
except UnqualifiedFontException as e:
|
| 82 |
print(f"SKIPPING Unqualified font: {e.font.path}")
|
| 83 |
+
runtime_exclusion_list.append(e.font.path)
|
| 84 |
with open(unqualified_log_file_name, "a+") as f:
|
| 85 |
f.write(f"{e.font.path}\n")
|
| 86 |
return
|