Upload 23 files

.gitattributes

@@ -56,3 +56,23 @@ python/examples/demo16.jpg filter=lfs diff=lfs merge=lfs -text
 python/examples/demo17.jpg filter=lfs diff=lfs merge=lfs -text
 python/examples/demo18.jpg filter=lfs diff=lfs merge=lfs -text
 python/examples/demo20.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo01.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo02.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo03.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo04.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo05.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo06.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo07.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo08.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo09.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo10.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo11.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo12.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo13.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo14.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo15.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo16.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo17.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo18.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo19.jpg filter=lfs diff=lfs merge=lfs -text
+examples/demo20.jpg filter=lfs diff=lfs merge=lfs -text

python/infer.py

@@ -1,7 +1,7 @@
 import argparse
 import cv2
 import numpy as np
-from axengine import InferenceSession
+import axengine as axe
 
 def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser()
@@ -15,7 +15,7 @@ def parse_args() -> argparse.Namespace:
         "--model",
         type=str,
         required=True,
-        help="Path to ONNX model.",
+        help="Path to axmodel model.",
     )
     
     return parser.parse_args()
@@ -25,12 +25,12 @@ def infer(img: str, model: str, viz: bool = False):
     img_raw = cv2.imread(img)
     image = cv2.cvtColor(img_raw, cv2.COLOR_BGR2RGB) 
     orig_h, orig_w = image.shape[:2]
-    image = cv2.resize(image, (518,518) )
+    image = cv2.resize(image, (518,518))
     image = image[None]
 
-    session = InferenceSession(path_or_bytes=model, providers= ['AxEngineExecutionProvider', 'AXCLRTExecutionProvider'])
+    session = axe.InferenceSession(model)
     
-    depth = session.run(output_names=["output"], input_feed={"input":image})[0]
+    depth = session.run(None, {"input": image})[0]
 
     depth = cv2.resize(depth[0, 0], (orig_w, orig_h))
     depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0