Googolxx / STF

Pytorch implementation of the paper "The Devil Is in the Details: Window-based Attention for Image Compression".
Apache License 2.0
161 stars 20 forks source link

Fixed reshape in decompressor #31

Closed MSNetrom closed 1 year ago

MSNetrom commented 1 year ago

There seems to be an error in a reshape in the decompressor of STF. The 'dequantization'-method will not run because of this reshaping.