Jian S
commited on
Commit
·
0033d20
1
Parent(s):
3d5b084
Updated README.md
Browse files
README.md
CHANGED
@@ -24,6 +24,8 @@ The repository is organized into two main folders:
|
|
24 |
|
25 |
### 2. **Models**
|
26 |
This folder contains complete models built with Kornia and exported to ONNX format. These models are designed for various computer vision tasks such as image processing, stereo vision, and more. Each model is accompanied by a detailed description of its purpose and how to use it.
|
|
|
|
|
27 |
|
28 |
## Getting Started
|
29 |
|
@@ -41,7 +43,7 @@ from kornia.onnx import ONNXSequential
|
|
41 |
onnx_seq = ONNXSequential(
|
42 |
"hf://operators/kornia.color.gray.RgbToGrayscale",
|
43 |
"hf://operators/kornia.geometry.transform.affwarp.Resize_512x512",
|
44 |
-
"YOUR_OWN_MODEL.onnx"
|
45 |
)
|
46 |
# Prepare some input data
|
47 |
input_data = np.random.randn(1, 3, 384, 512).astype(np.float32)
|
|
|
24 |
|
25 |
### 2. **Models**
|
26 |
This folder contains complete models built with Kornia and exported to ONNX format. These models are designed for various computer vision tasks such as image processing, stereo vision, and more. Each model is accompanied by a detailed description of its purpose and how to use it.
|
27 |
+
> [!Note]
|
28 |
+
> The detection models exported in this repo do not include a postprocessor, you may need to do it yourself or re-export the model by setting `confidence_filtering=True`. For example, `RTDETRDetectorBuilder.to_onnx(confidence_threshold=0.5, confidence_filtering=True)`.
|
29 |
|
30 |
## Getting Started
|
31 |
|
|
|
43 |
onnx_seq = ONNXSequential(
|
44 |
"hf://operators/kornia.color.gray.RgbToGrayscale",
|
45 |
"hf://operators/kornia.geometry.transform.affwarp.Resize_512x512",
|
46 |
+
"hf://models/kornia.models.detection.rtdetr_r18vd_640x640", # Or you may use "YOUR_OWN_MODEL.onnx"
|
47 |
)
|
48 |
# Prepare some input data
|
49 |
input_data = np.random.randn(1, 3, 384, 512).astype(np.float32)
|