Fix bw image processing

elisemercury / Duplicate-Image-Finder

difPy - Python package for finding duplicate or similar images within folders

https://difpy.readthedocs.io

MIT License

421 stars 65 forks source link

Closed ppizarror closed 2 years ago

ppizarror commented 2 years ago

Fixes #2, basically black-and-white images have only 2 dimensions if loaded by OpenCV, thus, the concatenate with the main matrix fails due to shape assertion. Now, if len(img.shape) is 2 the image is converted to three dimensions.
Format the code.
create_imgs_matrix now ensures directory exists. Now both dif.compare_images('folder') and dif.compare_images('folder/') works.

elisemercury commented 2 years ago

Thanks a lot for this! Merged into main.