File size: 1,006 Bytes
3a2845e |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
from huggingface_hub import hf_hub_download
from ultralytics import YOLO
import gradio as gr
from PIL import Image
import cv2
import numpy as np
import torch
path = hf_hub_download("Bingsu/adetailer", "person_yolov8s-seg.pt")
model = YOLO(path)
def infer(img: Image.Image):
img = "https://farm5.staticflickr.com/4139/4887614566_6b57ec4422_z.jpg"
output = model(img)
pred = output[0].plot()
pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
pred = Image.fromarray(pred)
return pred
inputs = gr.inputs.Image(type='pil', label="Original Image")
outputs = [
# gr.outputs.Image(type="numpy",label="Mask").style(height=600),
gr.outputs.Image(type="numpy",label="Mask"),
# gr.outputs.Image(type="numpy",label="Foreground"),
]
examples = [
['fox.jpg'],
['parrot.jpg']
]
gr.Interface(infer, inputs, outputs, title=f"anime seg", allow_flagging=False).launch(server_name=f"10.17.1.16", server_port=6019)
if __name__ == "__main__":
a = 1
|