| import os | |
| import sys | |
| import traceback | |
| import queue | |
| import threading | |
| import time | |
| import numpy as np | |
| import itertools | |
| from pathlib import Path | |
| from core import pathex | |
| from core import imagelib | |
| import cv2 | |
| import models | |
| from core.interact import interact as io | |
| def main(model_class_name, saved_models_path): | |
| model = models.import_model(model_class_name)( | |
| is_exporting=True, | |
| saved_models_path=saved_models_path, | |
| cpu_only=True) | |
| model.export_dfm () | |