initial commit
Browse files- app.py +1 -5
- example1_x2.jpg → examples/example1_x2.jpg +0 -0
- model.ort → models/model.ort +0 -0
- requirements.txt +1 -2
app.py
CHANGED
|
@@ -1,10 +1,6 @@
|
|
| 1 |
import numpy as np
|
| 2 |
import cv2
|
| 3 |
import onnxruntime
|
| 4 |
-
from glob import glob
|
| 5 |
-
import os
|
| 6 |
-
import gradio as gr
|
| 7 |
-
|
| 8 |
import gradio as gr
|
| 9 |
|
| 10 |
|
|
@@ -41,7 +37,7 @@ def convert_pil_to_cv2(image):
|
|
| 41 |
|
| 42 |
|
| 43 |
def main(image):
|
| 44 |
-
model_path = "
|
| 45 |
img = convert_pil_to_cv2(image)
|
| 46 |
if img.ndim == 2:
|
| 47 |
img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
|
|
|
|
| 1 |
import numpy as np
|
| 2 |
import cv2
|
| 3 |
import onnxruntime
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
import gradio as gr
|
| 5 |
|
| 6 |
|
|
|
|
| 37 |
|
| 38 |
|
| 39 |
def main(image):
|
| 40 |
+
model_path = "models/model.ort"
|
| 41 |
img = convert_pil_to_cv2(image)
|
| 42 |
if img.ndim == 2:
|
| 43 |
img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
|
example1_x2.jpg → examples/example1_x2.jpg
RENAMED
|
File without changes
|
model.ort → models/model.ort
RENAMED
|
File without changes
|
requirements.txt
CHANGED
|
@@ -1,3 +1,2 @@
|
|
| 1 |
numpy
|
| 2 |
-
onnxruntime
|
| 3 |
-
opencv-python
|
|
|
|
| 1 |
numpy
|
| 2 |
+
onnxruntime
|
|
|