let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add OpenCV2 model

Browse Source
This commit is contained in:
Qing
2022-09-25 21:27:12 +08:00
parent bc98ea256a
commit 35cbbd653c
5 changed files with 88 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lama_cleaner/model_manager.py
View File

@@ -4,9 +4,10 @@ from lama_cleaner.model.ldm import LDM
from lama_cleaner.model.mat import MAT
from lama_cleaner.model.sd import SD14
from lama_cleaner.model.zits import ZITS
from lama_cleaner.model.opencv2 import OpenCV2
from lama_cleaner.schema import Config
models = {"lama": LaMa, "ldm": LDM, "zits": ZITS, "mat": MAT, "fcf": FcF, "sd1.4": SD14}
models = {"lama": LaMa, "ldm": LDM, "zits": ZITS, "mat": MAT, "fcf": FcF, "sd1.4": SD14, "cv2": OpenCV2}
class ModelManager: