compressai_vision.datasets#