[FEATURE] Add support for other dtypes to Image in schema

🚨🚨 Feature Request

• A new implementation (Improvement, Extension)

Is your feature request related to a problem?

Currently, users are only able to use Image in schema with dtypes “uint8” and “uint16”. Using other integer and float types raises an error.

If your feature will improve HUB

More freedom for users to use different dtypes in Image.

Description of the possible solution

Modify hub.schema.Image to permit other integer and float dtypes. Also, add a check that ensures that in case the compressor is “png” and dtype other than “uint8” and “uint16” is specified, raise an error. Similarly, if the compressor is “jpeg” and a dtype other than “uint8” is specified, raise an error. The errors should be raised during the constructor call i.e. when _init_ is called.

Teachability, Documentation, Adoption, Migration Strategy After the fix, this code should work:-

schema_1 = {
    "image" : hub.schema.Image((500, 500, 3), "float64")
}

while this should raise an error

schema_2 = {
    "image" : hub.schema.Image((500, 500, 3), "float32", compressor="png")
}

[madhucharan]

Hi, I would like to know about all the expected dtypes to be included and any other checks needed for any other format other than the two checks mentioned (for png and jpeg)

[mynameisvinn]

Closing due to inactivity.