OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Actually load rgb_alpha images. (#2942) 

This should've been added in #2925.

The only doubt I have around the function name:

- `load_rgb_alpha_image`: following the `rgb_alpha_pixel` in dlib
- `load_rgba_image`: shorter and maybe nicer?
This commit is contained in:
Adrià Arrufat 2024-04-04 08:26:02 +09:00 committed by GitHub
parent 8edfa542b8
commit ca7fd9d1ad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tools/python/src/numpy_returns.cpp
View File

@ -20,7 +20,7 @@ numpy_image<rgb_pixel> load_rgb_image (const std::string &path)
return img; return img;
} }
numpy_image<rgb_pixel> load_rgb_alpha_image (const std::string &path) numpy_image<rgb_alpha_pixel> load_rgb_alpha_image (const std::string &path)
{ {
numpy_image<rgb_alpha_pixel> img; numpy_image<rgb_alpha_pixel> img;
load_image(img, path); load_image(img, path);
@ -166,6 +166,11 @@ void bind_numpy_returns(py::module &m)
py::arg("filename") py::arg("filename")
); );
m.def("load_rgb_alpha_image", &load_rgb_alpha_image,
"Takes a path and returns a numpy array (RGBA) containing the image",
py::arg("filename")
);
m.def("load_grayscale_image", &load_grayscale_image, m.def("load_grayscale_image", &load_grayscale_image,
"Takes a path and returns a numpy array containing the image, as an 8bit grayscale image.", "Takes a path and returns a numpy array containing the image, as an 8bit grayscale image.",
py::arg("filename") py::arg("filename")