Skip to content

Commit

Permalink
Actually load rgb_alpha images. (#2942)
Browse files Browse the repository at this point in the history
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?
  • Loading branch information
arrufat committed Apr 3, 2024
1 parent 8edfa54 commit ca7fd9d
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion tools/python/src/numpy_returns.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ numpy_image<rgb_pixel> load_rgb_image (const std::string &path)
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;
load_image(img, path);
Expand Down Expand Up @@ -166,6 +166,11 @@ void bind_numpy_returns(py::module &m)
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,
"Takes a path and returns a numpy array containing the image, as an 8bit grayscale image.",
py::arg("filename")
Expand Down

0 comments on commit ca7fd9d

Please sign in to comment.