kornia
/

ONNX_models

Model card Files Files and versions

Jian S commited on Sep 9, 2024

Commit

b00a6f9

·

1 Parent(s): 0033d20

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -41,8 +41,7 @@ import numpy as np
 from kornia.onnx import ONNXSequential
 # Chain ONNX models from HuggingFace repo and your own local model together
 onnx_seq = ONNXSequential(
-    "hf://operators/kornia.color.gray.RgbToGrayscale",
-    "hf://operators/kornia.geometry.transform.affwarp.Resize_512x512",
     "hf://models/kornia.models.detection.rtdetr_r18vd_640x640",  # Or you may use "YOUR_OWN_MODEL.onnx"
 )
 # Prepare some input data

 from kornia.onnx import ONNXSequential
 # Chain ONNX models from HuggingFace repo and your own local model together
 onnx_seq = ONNXSequential(
+    "hf://operators/kornia.geometry.transform.flips.Hflip",
     "hf://models/kornia.models.detection.rtdetr_r18vd_640x640",  # Or you may use "YOUR_OWN_MODEL.onnx"
 )
 # Prepare some input data