Allow "tiff" and more extensions in DetectionDataset.from_yolo function

[patel-zeel]

Description

Addresses #1554 as discussed with @LinasKo.

Highlights

• Allows any image extension in DetectionDataset.from_yolo.
  Tested extensions (listed on Ultralytics predict page):
  • bmp
  • jpg
  • mpo
  • png
  • tif
  • webp
  • pfm - Pillow could not read pfm images, cv2 is able to read them. I saw here that even Ultralytics uses Pillow to read images. In that case, I am not sure if Ultralytics can "really" support pfm format, as their documentation suggests.
  • dng
• Image must be of "RGB" type, raise ValueError otherwise.

List any dependencies that are required for this change.

• Pillow

Type of change

Please delete options that are not relevant.

• New feature (non-breaking change which adds functionality)

How has this change been tested, please provide a testcase or example of how you tested the change?

The changes are tested with this colab notebook.

Test Case

100 images of each extension were generated along with their corresponding dummy labels and dummy data.yml. Test run checks if the function works for a particular extension and computes time taken to load the dataset.

Time taken (in seconds) to run the test case

• Columns 0,1,2,3,4 are seed numbers.
• -1.11 value indicates that supervision doesn't support that extension yet.

Result for supervision version 0.24.0, which uses cv2 to check the shape of images.

	0	1	2	3	4	Mean	Std
bmp	-1.11	-1.11	-1.11	-1.11	-1.11	-1.11	0.00
jpg	0.98	1.16	0.85	0.76	0.77	0.90	0.15
mpo	-1.11	-1.11	-1.11	-1.11	-1.11	-1.11	0.00
png	0.58	0.56	0.42	0.4	0.41	0.48	0.08
tif	-1.11	-1.11	-1.11	-1.11	-1.11	-1.11	0.00
webp	-1.11	-1.11	-1.11	-1.11	-1.11	-1.11	0.00
dng	-1.11	-1.11	-1.11	-1.11	-1.11	-1.11	0.00

Result for this PR, which uses Pillow to check the shape and type (RGB or not) of images.

	0	1	2	3	4	Mean	Std
bmp	0.04	0.02	0.02	0.02	0.03	0.03	0.01
jpg	0.03	0.02	0.03	0.02	0.02	0.03	0.00
mpo	0.04	0.04	0.06	0.04	0.04	0.04	0.01
png	0.02	0.02	0.04	0.02	0.02	0.02	0.01
tif	0.05	0.06	0.08	0.05	0.05	0.06	0.01
webp	0.10	0.11	0.11	0.09	0.11	0.10	0.00
dng	0.06	0.06	0.05	0.06	0.06	0.06	0.00

Docs

• Docs updated? What were the changes: Not changed, please suggest if we need changes in the docs

[LinasKo]

You never cease to surprise, @patel-zeel; such a thorough analysis! ⭐

Adding this to the back of my review backlog, but I already know it will be a delight.

[patel-zeel]

Thank you for your kind words, @LinasKo. It's a pleasure to help improve a widely used library. Looking forward to your feedback!