File size: 874 Bytes
9dce458 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import os
import numpy as np
from abc import abstractmethod
from ..utils import InfererModule, ModelWrapper
class CommonInpainter(InfererModule):
async def inpaint(self, image: np.ndarray, mask: np.ndarray, inpainting_size: int = 1024, verbose: bool = False) -> np.ndarray:
return await self._inpaint(image, mask, inpainting_size, verbose)
@abstractmethod
async def _inpaint(self, image: np.ndarray, mask: np.ndarray, inpainting_size: int = 1024, verbose: bool = False) -> np.ndarray:
pass
class OfflineInpainter(CommonInpainter, ModelWrapper):
_MODEL_SUB_DIR = 'inpainting'
async def _inpaint(self, *args, **kwargs):
return await self.infer(*args, **kwargs)
@abstractmethod
async def _infer(self, image: np.ndarray, mask: np.ndarray, inpainting_size: int = 1024, verbose: bool = False) -> np.ndarray:
pass
|