Files
2026-07-13 12:35:17 +08:00

26 lines
593 B
Python

import importlib
import os
import sys
from pathlib import Path
root_dir = Path(__file__).parent.parent.resolve()
os.environ['PYTHONPATH'] = str(root_dir)
sys.path.insert(0, str(root_dir))
from utils.hparams import set_hparams, hparams
set_hparams()
def binarize():
binarizer_cls = hparams["binarizer_cls"]
pkg = ".".join(binarizer_cls.split(".")[:-1])
cls_name = binarizer_cls.split(".")[-1]
binarizer_cls = getattr(importlib.import_module(pkg), cls_name)
print("| Binarizer: ", binarizer_cls)
binarizer_cls().process()
if __name__ == '__main__':
binarize()