Jian S commited on
Commit
0033d20
·
1 Parent(s): 3d5b084

Updated README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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)