mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2024-03-22 13:10:55 +08:00
23 lines
525 B
Python
23 lines
525 B
Python
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 ()
|