From cd32c65e9d010c87dc32c4191786e3f7194af26c Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Mon, 9 May 2022 15:50:03 +0200
Subject: [PATCH] Fix docs
---
bob/io/base/__init__.py | 5 +++--
doc/py_api.rst | 13 +++++++++++--
2 files changed, 14 insertions(+), 4 deletions(-)
diff --git a/bob/io/base/__init__.py b/bob/io/base/__init__.py
index 3e4da95..fd99f4a 100644
--- a/bob/io/base/__init__.py
+++ b/bob/io/base/__init__.py
@@ -5,8 +5,6 @@ import h5py
import imageio
import numpy as np
-from ..image import to_bob, to_matplotlib
-
logger = logging.getLogger(__name__)
import os
@@ -100,6 +98,7 @@ def open_file(filename):
return np.array(f[key])
elif extension in image_extensions:
+ from ..image import to_bob
img = imageio.imread(filename)
@@ -143,6 +142,8 @@ def write_file(filename, data, format="pillow"):
f["array"] = data
elif extension in image_extensions:
# Pillow is the format with the best support for all image formats
+ from ..image import to_matplotlib
+
imageio.imwrite(filename, to_matplotlib(data), format=format)
else:
raise RuntimeError(f"Unknown file extension: {extension}")
diff --git a/doc/py_api.rst b/doc/py_api.rst
index 2373795..406ac91 100644
--- a/doc/py_api.rst
+++ b/doc/py_api.rst
@@ -12,7 +12,16 @@ Functions
---------
.. autosummary::
- bob.io.base.load
bob.io.base.save
- bob.io.base.create_directories_safe
+ bob.io.base.load
bob.io.base.vstack_features
+ bob.io.image.to_matplotlib
+ bob.io.image.to_bob
+ bob.io.image.bob_to_pillow
+ bob.io.image.pillow_to_bob
+ bob.io.image.opencvbgr_to_bob
+ bob.io.image.bob_to_opencvbgr
+ bob.io.image.imshow
+
+.. automodule:: bob.io.base
+.. automodule:: bob.io.image
--
GitLab