fix python bindings

.github/workflows/python_test.yml

@@ -2,17 +2,10 @@ name: Python Test
 
 on:
   pull_request:
-    paths:
-      - Cargo.lock
-      - kornia-py/**
-      - .github/workflows/test-python.yml
+    - main
   push:
     branches:
       - main
-    paths:
-      - Cargo.lock
-      - kornia-py/**
-      - .github/workflows/test-python.yml
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}

.gitignore

@@ -1,4 +1,4 @@
 .vscode/
-.venv/
+**venv/
 target/
-docker/
+docker/

kornia-py/benchmark/bench_io.py

@@ -21,6 +21,7 @@ def read_image_pillow(image_path: str) -> None:
 def read_image_kornia(image_path: str) -> None:
     return kornia_rs.read_image_jpeg(image_path)
 
+
 def read_image_tensorflow(image_path: str) -> None:
     return tf.keras.utils.load_img(image_path)
 

kornia-py/src/image.rs

@@ -9,12 +9,12 @@ pub type PyImage = Py<PyArray3<u8>>;
 
 /// Trait to convert an image to a PyImage (3D numpy array of u8)
 pub trait ToPyImage {
-    fn to_pyimage(&self) -> PyImage;
+    fn to_pyimage(self) -> PyImage;
 }
 
 impl<const CHANNELS: usize> ToPyImage for kornia_rs::image::Image<u8, CHANNELS> {
-    fn to_pyimage(&self) -> PyImage {
-        Python::with_gil(|py| self.data.to_pyarray(py).to_owned())
+    fn to_pyimage(self) -> PyImage {
+        Python::with_gil(|py| self.data().to_pyarray(py).to_owned())
     }
 }