Duplicate from sayakpaul/cartoonizer-demo-onnx

Co-authored-by: Sayak Paul <[email protected]>

.gitattributes

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Cartoonizer Demo ONNX
+emoji: 🗻
+colorFrom: green
+colorTo: gray
+sdk: gradio
+sdk_version: 3.1.4
+app_file: app.py
+pinned: false
+license: apache-2.0
+duplicated_from: sayakpaul/cartoonizer-demo-onnx
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,54 @@
+import os
+
+import cv2
+import gradio as gr
+import numpy as np
+import onnxruntime as ort
+from PIL import Image
+
+_sess_options = ort.SessionOptions()
+_sess_options.intra_op_num_threads = os.cpu_count()
+MODEL_SESS = ort.InferenceSession(
+    "cartoonizer.onnx", _sess_options, providers=["CPUExecutionProvider"]
+)
+
+
+def preprocess_image(image: Image) -> np.ndarray:
+    image = np.array(image)
+    image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+
+    h, w, c = np.shape(image)
+    if min(h, w) > 720:
+        if h > w:
+            h, w = int(720 * h / w), 720
+        else:
+            h, w = 720, int(720 * w / h)
+    image = cv2.resize(image, (w, h), interpolation=cv2.INTER_AREA)
+    h, w = (h // 8) * 8, (w // 8) * 8
+    image = image[:h, :w, :]
+    image = image.astype(np.float32) / 127.5 - 1
+    return np.expand_dims(image, axis=0)
+
+
+def inference(image: np.ndarray) -> Image:
+    image = preprocess_image(image)
+    results = MODEL_SESS.run(None, {"input_photo:0": image})
+    output = (np.squeeze(results[0]) + 1.0) * 127.5
+    output = np.clip(output, 0, 255).astype(np.uint8)
+    output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
+    return Image.fromarray(output)
+
+
+title = "Generate cartoonized images"
+article = "Demo of CartoonGAN model (https://systemerrorwang.github.io/White-box-Cartoonization/). \nDemo image is from https://unsplash.com/photos/f0SgAs27BYI."
+
+iface = gr.Interface(
+    inference,
+    inputs=gr.inputs.Image(type="pil", label="Input Image"),
+    outputs="image",
+    title=title,
+    article=article,
+    allow_flagging="never",
+    examples=[["mountain.jpeg"]],
+)
+iface.launch()

cartoonizer.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c035adb7ce7590680fca3d82941ed5d1f6bcb25a6f61ac3fea06dcfb656c325a
+size 5886605

requirements.txt

@@ -0,0 +1,3 @@
+onnxruntime
+numpy
+opencv-python