Hugging Face's logo

kornia
/
ONNX_models

ONNX
English
kornia
Model card Files Community
shijianjian commited on
Commit
45eabf5
•
1 Parent(s): 3d9c988

init operators

Browse files
Files changed (33) hide show
  1. .gitattributes +35 -0
  2. README.md +46 -0
  3. banner.jpeg +0 -0
  4. operators/kornia.color.gray.BgrToGrayscale.onnx +3 -0
  5. operators/kornia.color.gray.GrayscaleToRgb.onnx +3 -0
  6. operators/kornia.color.gray.RgbToGrayscale.onnx +3 -0
  7. operators/kornia.color.hls.HlsToRgb.onnx +3 -0
  8. operators/kornia.color.hls.RgbToHls.onnx +3 -0
  9. operators/kornia.color.hsv.HsvToRgb.onnx +3 -0
  10. operators/kornia.color.hsv.RgbToHsv.onnx +3 -0
  11. operators/kornia.color.lab.LabToRgb.onnx +3 -0
  12. operators/kornia.color.lab.RgbToLab.onnx +3 -0
  13. operators/kornia.color.luv.LuvToRgb.onnx +3 -0
  14. operators/kornia.color.luv.RgbToLuv.onnx +3 -0
  15. operators/kornia.color.rgb.BgrToRgb.onnx +3 -0
  16. operators/kornia.color.rgb.LinearRgbToRgb.onnx +3 -0
  17. operators/kornia.color.rgb.RgbToBgr.onnx +3 -0
  18. operators/kornia.color.rgb.RgbToLinearRgb.onnx +3 -0
  19. operators/kornia.color.sepia.Sepia.onnx +3 -0
  20. operators/kornia.color.xyz.RgbToXyz.onnx +3 -0
  21. operators/kornia.color.xyz.XyzToRgb.onnx +3 -0
  22. operators/kornia.color.ycbcr.RgbToYcbcr.onnx +3 -0
  23. operators/kornia.color.ycbcr.YcbcrToRgb.onnx +3 -0
  24. operators/kornia.color.yuv.RgbToYuv.onnx +3 -0
  25. operators/kornia.color.yuv.YuvToRgb.onnx +3 -0
  26. operators/kornia.geometry.transform.affwarp.Resize_256x256.onnx +3 -0
  27. operators/kornia.geometry.transform.affwarp.Resize_384x384.onnx +3 -0
  28. operators/kornia.geometry.transform.affwarp.Resize_512x512.onnx +3 -0
  29. operators/kornia.geometry.transform.affwarp.Resize_576x1024.onnx +3 -0
  30. operators/kornia.geometry.transform.affwarp.Resize_640x640.onnx +3 -0
  31. operators/kornia.geometry.transform.flips.Hflip.onnx +3 -0
  32. operators/kornia.geometry.transform.flips.Rot180.onnx +3 -0
  33. operators/kornia.geometry.transform.flips.Vflip.onnx +3 -0
.gitattributes CHANGED
@@ -33,3 +33,38 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ operators/kornia.color.gray.BgrToGrayscale.onnx filter=lfs diff=lfs merge=lfs -text
37
+ operators/kornia.color.gray.GrayscaleToRgb.onnx filter=lfs diff=lfs merge=lfs -text
38
+ operators/kornia.color.hls.RgbToHls.onnx filter=lfs diff=lfs merge=lfs -text
39
+ operators/kornia.color.rgb.LinearRgbToRgb.onnx filter=lfs diff=lfs merge=lfs -text
40
+ operators/kornia.color.rgb.RgbToBgr.onnx filter=lfs diff=lfs merge=lfs -text
41
+ operators/kornia.color.gray.RgbToGrayscale.onnx filter=lfs diff=lfs merge=lfs -text
42
+ operators/kornia.color.yuv.YuvToRgb.onnx filter=lfs diff=lfs merge=lfs -text
43
+ operators/kornia.color.luv.LuvToRgb.onnx filter=lfs diff=lfs merge=lfs -text
44
+ operators/kornia.color.sepia.Sepia.onnx filter=lfs diff=lfs merge=lfs -text
45
+ operators/kornia.color.yuv.RgbToYuv.onnx filter=lfs diff=lfs merge=lfs -text
46
+ operators/kornia.geometry.transform.affwarp.Resize_256X256.onnx filter=lfs diff=lfs merge=lfs -text
47
+ operators/kornia.geometry.transform.flips.Hflip.onnx filter=lfs diff=lfs merge=lfs -text
48
+ operators/kornia.color.hsv.RgbToHsv.onnx filter=lfs diff=lfs merge=lfs -text
49
+ operators/kornia.color.lab.LabToRgb.onnx filter=lfs diff=lfs merge=lfs -text
50
+ operators/kornia.color.luv.RgbToLuv.onnx filter=lfs diff=lfs merge=lfs -text
51
+ operators/kornia.geometry.transform.affwarp.Resize_512X512.onnx filter=lfs diff=lfs merge=lfs -text
52
+ operators/kornia.geometry.transform.affwarp.Resize_576X1024.onnx filter=lfs diff=lfs merge=lfs -text
53
+ operators/kornia.geometry.transform.affwarp.Resize_640X640.onnx filter=lfs diff=lfs merge=lfs -text
54
+ operators/kornia.color.hsv.HsvToRgb.onnx filter=lfs diff=lfs merge=lfs -text
55
+ operators/kornia.color.rgb.RgbToLinearRgb.onnx filter=lfs diff=lfs merge=lfs -text
56
+ operators/kornia.geometry.transform.flips.Rot180.onnx filter=lfs diff=lfs merge=lfs -text
57
+ operators/kornia.geometry.transform.flips.Vflip.onnx filter=lfs diff=lfs merge=lfs -text
58
+ operators/kornia.color.hls.HlsToRgb.onnx filter=lfs diff=lfs merge=lfs -text
59
+ operators/kornia.color.lab.RgbToLab.onnx filter=lfs diff=lfs merge=lfs -text
60
+ operators/kornia.color.rgb.BgrToRgb.onnx filter=lfs diff=lfs merge=lfs -text
61
+ operators/kornia.color.xyz.RgbToXyz.onnx filter=lfs diff=lfs merge=lfs -text
62
+ operators/kornia.color.xyz.XyzToRgb.onnx filter=lfs diff=lfs merge=lfs -text
63
+ operators/kornia.color.ycbcr.YcbcrToRgb.onnx filter=lfs diff=lfs merge=lfs -text
64
+ operators/kornia.geometry.transform.affwarp.Resize_384X384.onnx filter=lfs diff=lfs merge=lfs -text
65
+ operators/kornia.color.ycbcr.RgbToYcbcr.onnx filter=lfs diff=lfs merge=lfs -text
66
+ operators/kornia.geometry.transform.affwarp.Resize_256x256.onnx filter=lfs diff=lfs merge=lfs -text
67
+ operators/kornia.geometry.transform.affwarp.Resize_576x1024.onnx filter=lfs diff=lfs merge=lfs -text
68
+ operators/kornia.geometry.transform.affwarp.Resize_512x512.onnx filter=lfs diff=lfs merge=lfs -text
69
+ operators/kornia.geometry.transform.affwarp.Resize_640x640.onnx filter=lfs diff=lfs merge=lfs -text
70
+ operators/kornia.geometry.transform.affwarp.Resize_384x384.onnx filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Kornia ONNX Repository
2
+
3
+ Welcome to the **Kornia ONNX Repository**! This repository hosts ONNX files for models and operators related to the Kornia computer vision library. Here, you'll find pre-trained models, custom operators, and resources to accelerate your machine learning and computer vision workflows using ONNX format.
4
+
5
+ ![Kornia ONNX](./banner.jpeg)
6
+
7
+
8
+ ## Repository Structure
9
+
10
+ By default, we include ONNX models with dynamic shapes. If the model is exported with a specific input shape, it would be noted in the filename such as `XXX_512x512.onnx`.
11
+
12
+ The repository is organized into two main folders:
13
+
14
+ ### 1. **Operators**
15
+ This folder contains various ONNX operators used within the Kornia library. These operators are designed to be modular and can be integrated into different models or workflows. Each operator comes with a brief description and example usage.
16
+
17
+ ### 2. **Models**
18
+ 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.
19
+
20
+ ## Getting Started
21
+
22
+ Install Kornia library:
23
+ ```shell
24
+ pip install -U kornia
25
+ ```
26
+
27
+ To chain many operators or models into one graph, follow these steps:
28
+
29
+ ```python
30
+ import numpy as np
31
+ from kornia.onnx import ONNXSequential
32
+ # Initialize ONNXSequential with two models
33
+ onnx_seq = ONNXSequential("kornia.color.gray.RgbToGrayscale.onnx", "kornia.geometry.transform.flips.Hflip.onnx")
34
+ # Prepare some input data
35
+ input_data = np.random.randn(1, 3, 224, 224).astype(np.float32)
36
+ # Perform inference
37
+ outputs = onnx_seq(input_data)
38
+ # Print the model outputs
39
+ print(outputs)
40
+ ```
41
+
42
+ ## Contribution
43
+
44
+ We welcome contributions! If you'd like to add your own ONNX operators or models, feel free to open a pull request. Please ensure that you include relevant documentation and example usage for any new additions.
45
+
46
+ Thank you for using the Kornia ONNX Repository! Feel free to explore and contribute to the growing collection of operators and models.
banner.jpeg ADDED
operators/kornia.color.gray.BgrToGrayscale.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdc2bd223edacd1aef81fa76d5e5cd25de7cde0cb31ce5a3d1eb4eac46170968
3
+ size 2452
operators/kornia.color.gray.GrayscaleToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfb46ea9e7b8d0f74b6691df4b2f041e7216c9149bf4ad2e97d0dd493846e79e
3
+ size 203
operators/kornia.color.gray.RgbToGrayscale.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16271c02e84d09dd4162ad5714ba7d132ba35d1a8de9e957ac230e62cd463678
3
+ size 1974
operators/kornia.color.hls.HlsToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:928c65c3486626f7d3532979e51c004b037b209214cdbf2d9e8912bb0500f452
3
+ size 2579
operators/kornia.color.hls.RgbToHls.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3448e254db698253975caf50e205b19f009f9dd96e5139a65b726d4e5a4fb817
3
+ size 260
operators/kornia.color.hsv.HsvToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fe0fdcec8fce9a50b1f8c81606da8c7ccc60a2d515fe424838273f76f76406d
3
+ size 7293
operators/kornia.color.hsv.RgbToHsv.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ff880ead7cadc20121f4794ecd6cb5147f09c09baf7d6e8f18882fc6a990b82
3
+ size 5230
operators/kornia.color.lab.LabToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a239c5443f805363a5f99e1edffd5c2f6a0fbc0eecf4d74b998cbc63f146606
3
+ size 6673
operators/kornia.color.lab.RgbToLab.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:687930cfc3aa2b2cea0ac89517e03ba6a6c8a493505e000a597add4b308376bc
3
+ size 6289
operators/kornia.color.luv.LuvToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6304ff98dea8fa6199dcfd8435a9e1d82bd2727f6b826e4a471169fe1f2a2529
3
+ size 8172
operators/kornia.color.luv.RgbToLuv.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f2f0d562158de59742ccfcd6bdf0db6726387665490b6d7add65af5ab7b9599
3
+ size 7136
operators/kornia.color.rgb.BgrToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:395f30500617c4950615961cdd8da755c7074a43d270adf66262d21a607b58bc
3
+ size 547
operators/kornia.color.rgb.LinearRgbToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e328a586ba8c9b17fc9da01aaf62a66102fcd8e34964ac73313083d18fc2110b
3
+ size 1005
operators/kornia.color.rgb.RgbToBgr.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:395f30500617c4950615961cdd8da755c7074a43d270adf66262d21a607b58bc
3
+ size 547
operators/kornia.color.rgb.RgbToLinearRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4239d80f60ac2d98decd5d1a1b1d395a86f0ff86c3fafc07ae41eeb4d075736e
3
+ size 865
operators/kornia.color.sepia.Sepia.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f048d42ab2e3346697c076e5a40ef1417d58c16c69e349ab1f1c5c8a1ca13839
3
+ size 3659
operators/kornia.color.xyz.RgbToXyz.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9dda1c72594abb052c1bdde8e388f4346f51a618c1e8cf2df06e5a82e2345ed9
3
+ size 2875
operators/kornia.color.xyz.XyzToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86fb4ddf5740b33e572a73816dbe72bfc736bc490da8417670d2b8161cac7629
3
+ size 2875
operators/kornia.color.ycbcr.RgbToYcbcr.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0d348652762db4da9087155a0b27f13da30b8a7cec4921c1329ca6e5d863a1c
3
+ size 2441
operators/kornia.color.ycbcr.YcbcrToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e4caec9b1866d7ff5e3a7d0e0926351f765a6992d1ce43901575cd836fe84b2
3
+ size 2533
operators/kornia.color.yuv.RgbToYuv.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96f6f700aba11206a83bf2c1d08dc04e99cd606bf51b6d74b29999ca4dd9d664
3
+ size 2869
operators/kornia.color.yuv.YuvToRgb.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1633286a99ced27f9352ccfd6b17dbe52862898ed3b3e348675465c6b7ded6cd
3
+ size 2016
operators/kornia.geometry.transform.affwarp.Resize_256x256.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f67b6b21a08debfd1cb8fd544a4dcc96183812c51d41f3dd3f3c6fa26600a45e
3
+ size 182
operators/kornia.geometry.transform.affwarp.Resize_384x384.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4362fe8b10f2ee94951c15f8a903b00761f58ef737866b6f387c01b578aac418
3
+ size 868
operators/kornia.geometry.transform.affwarp.Resize_512x512.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a1921e97e7d8759444e74ba7ad70059d03fd33db378809a923c83709d0375681
3
+ size 868
operators/kornia.geometry.transform.affwarp.Resize_576x1024.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b82df37fda04ef18510bfc335886126f463d61060aa5f057907ea3fdd6ce69e3
3
+ size 868
operators/kornia.geometry.transform.affwarp.Resize_640x640.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21dfaa0791861d5efb961f0fc6df8e58c814bb48fc142046d7850f91eefc8b4f
3
+ size 868
operators/kornia.geometry.transform.flips.Hflip.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cb5dabd6d0a2b4a935ee97ae3752870427ab63f3b9d3668f8d5431209d02482
3
+ size 557
operators/kornia.geometry.transform.flips.Rot180.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94f769eb4cae1c7d3cdbfbf61fdfc61c66840ce3ee1654d3dac37e1d74524349
3
+ size 589
operators/kornia.geometry.transform.flips.Vflip.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e663d09700fdcdb34aeb3dd1f3c0ba02dfeb8516e12d055367c73abfca84bbc
3
+ size 557