|
from transformers import Pipeline , AutoModel, AutoImageProcessor, pipeline |
|
from PIL import Image |
|
from loadimg import load_img |
|
import torch |
|
REPO_NAME = "p1atdev/MangaLineExtraction-hf" |
|
|
|
class MangaLineExtractionPipe(Pipeline): |
|
def __init__(self,**kwargs): |
|
self.processor = AutoImageProcessor.from_pretrained("p1atdev/MangaLineExtraction-hf", trust_remote_code=True) |
|
Pipeline.__init__(self,**kwargs) |
|
self.model.eval() |
|
def _sanitize_parameters(kwargs): |
|
return {}, {} , {} |
|
def preprocess(self,image): |
|
|
|
image = load_img(image, output_type = "pil").convert("RGB") |
|
processed = self.processor(image, return_tensors="pt") |
|
return processed.pixel_values |
|
def _forward(self,inputs): |
|
return self.model(inputs) |
|
def postprocess(self,outputs): |
|
return Image.fromarray(outputs.pixel_values[0].numpy().astype("uint8"), mode="L") |
|
|
