Add support for .webp images. (#153)

* Add support for `.webp` images. WebP images are getting increasingly more common, and Pillow supports this format. Adding it to the list of valid formats is the only change required to make this work. * Add `.webp` to supported formats in docs * Convert test image to WEBP and add to tests
idealo · Jan 10, 2023 · a2b0837 · a2b0837
1 parent cd7a946
commit a2b0837
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/imagededup/utils/image_utils.py b/imagededup/utils/image_utils.py
@@ -7,7 +7,7 @@
 from imagededup.utils.logger import return_logger
 
 
-IMG_FORMATS = ['JPEG', 'PNG', 'BMP', 'MPO', 'PPM', 'TIFF', 'GIF']
+IMG_FORMATS = ['JPEG', 'PNG', 'BMP', 'MPO', 'PPM', 'TIFF', 'GIF', 'WEBP']
 logger = return_logger(__name__)
 
 

diff --git a/mkdocs/docs/user_guide/encoding_generation.md b/mkdocs/docs/user_guide/encoding_generation.md
@@ -45,7 +45,7 @@ The 'method-name' corresponds to one of the deduplication methods available and
 
 - If an image in the image directory can't be loaded, no encodings are generated for the image. Hence, there is no entry 
 for the image in the returned encodings dictionary.
-- Supported image formats: 'JPEG', 'PNG', 'BMP', 'MPO', 'PPM', 'TIFF', 'GIF', 'SVG', 'PGM', 'PBM'.
+- Supported image formats: 'JPEG', 'PNG', 'BMP', 'MPO', 'PPM', 'TIFF', 'GIF', 'SVG', 'PGM', 'PBM', 'WEBP'.
 
 #### Examples
 
@@ -74,7 +74,7 @@ array if cnn is used.
 #### Considerations
 
 - If the image can't be loaded, no encodings are generated for the image and *None* is returned.
-- Supported image formats: 'JPEG', 'PNG', 'BMP', 'MPO', 'PPM', 'TIFF', 'GIF', 'SVG', 'PGM', 'PBM'.
+- Supported image formats: 'JPEG', 'PNG', 'BMP', 'MPO', 'PPM', 'TIFF', 'GIF', 'SVG', 'PGM', 'PBM', 'WEBP'.
 
 #### Examples
 

diff --git a/tests/data/formats_images/ukbench09380.webp b/tests/data/formats_images/ukbench09380.webp
diff --git a/tests/test_cnn.py b/tests/test_cnn.py
@@ -92,6 +92,7 @@ def test__get_cnn_features_batch(cnn):
         'ukbench09380.jpeg',
         'ukbench09380.png',
         'ukbench09380.svg',
+        'ukbench09380.webp',
     ]
 
     assert list(sorted(result.keys(), key=str.lower)) == expected_predicted_files
@@ -204,6 +205,7 @@ def test_encode_images(cnn):
         'ukbench09380.jpeg',
         'ukbench09380.png',
         'ukbench09380.svg',
+        'ukbench09380.webp',
     ]
 
     assert list(sorted(result.keys(), key=str.lower)) == expected_predicted_files
@@ -225,6 +227,7 @@ def test_encode_images(cnn):
         'ukbench09380.jpeg',
         'ukbench09380.png',
         'ukbench09380.svg',
+        'ukbench09380.webp',
     ]
 
     assert list(sorted(result.keys(), key=str.lower)) == expected_predicted_files