如何将 IOPAint 装入自定义类

我安装了IOPAint。 迄今为止,这些特点的工作完美无缺:

启动网络服务器

def run iopaint single(图像 path, mask path, 输出 path, model='lama', 设备='cpu') : ""单相相像面具对上运行的IOPAint"". cmd=[ "画","跑", "-图像",图像-路径, "-面具",口罩-路径, ——输出,输出 路径, '--型',型号, '-device',设备 [ . ] 尝试 : results=sub process.run(cmd,抓取 output=True,文本=True,检查=True). 返回 True,结果。 静态 除了子进程。 名为程序错误 : 返回 False, f"错误 : {e.stderr}"

将iopaint模型装入一个类的变量并调用过程的正确方法是什么? 沿着这条线

将数字导入为 np 导入 cv2 导入 iopaint

类印记 : def init (自,型号:str='lama',设备:str='cuda'): 自我.engine = iopaint.get model(模型,设备) 自我。模型 = 模型 自我.device = 设备

def inpaint(自,图像:np.ndarray,口罩:np.ndarray) - >np.ndarray: 做点什么 返回自定义.engine.inpaint(图像、口罩)

如果... 名字... "... main...": 因帕inter = 因帕inter () img = cv2. imread ('path/ to/ image') 口罩 = cv2.imread ('path/ to/mask', 0) ; reslt = inpainter.inpaint (img, 口罩)