CompressAI icon indicating copy to clipboard operation
CompressAI copied to clipboard

where is the source code of self.entropy_bottleneck.compress()?

Open JF-Lee opened this issue 2 years ago • 2 comments

Thanks of your excellent work! Now I wanna to deploy FactorizedPrior on embedded device, and I found EntropyBottleneck in it, but I can not find .compress() and decompress(), so can anyone tell me about that? Thanks again!

JF-Lee avatar Mar 12 '23 11:03 JF-Lee

EntropyBottleneck's compress/decompress:

https://github.com/InterDigitalInc/CompressAI/blob/ee91d536bd934fc1f8b1532f78db4c94072ae26d/compressai/entropy_models/entropy_models.py#L524-L537

Or are you referring to a "frozen"/"serialized"/"exported" model not having access to these functions?

YodaEmbedding avatar Mar 13 '23 09:03 YodaEmbedding

EntropyBottleneck's compress/decompress:

https://github.com/InterDigitalInc/CompressAI/blob/ee91d536bd934fc1f8b1532f78db4c94072ae26d/compressai/entropy_models/entropy_models.py#L524-L537

Or are you referring to a "frozen"/"serialized"/"exported" model not having access to these functions?

=================================================================== Thanks for your reply! I want to access to these functions, but I can not find that anywhere, so can you tell me where I can find that?

JF-Lee avatar Mar 13 '23 09:03 JF-Lee