US 12,225,205 B2
Block-wise entropy coding method in neural image compression
Sheng Lin, San Jose, CA (US); Wei Jiang, Sunnyvale, CA (US); Shan Liu, San Jose, CA (US); and Wei Wang, Palo Alto, CA (US)
Assigned to TENCENT AMERICA LLC, Palo Alto, CA (US)
Filed by TENCENT AMERICA LLC, Palo Alto, CA (US)
Filed on May 26, 2022, as Appl. No. 17/825,575.
Claims priority of provisional application 63/212,469, filed on Jun. 18, 2021.
Prior Publication US 2022/0408098 A1, Dec. 22, 2022
Int. Cl. H04N 19/149 (2014.01); H04N 19/176 (2014.01); H04N 19/60 (2014.01); H04N 19/91 (2014.01)
CPC H04N 19/149 (2014.11) [H04N 19/176 (2014.11); H04N 19/60 (2014.11); H04N 19/91 (2014.11)] 8 Claims
OG exemplary drawing
 
1. A method performed by at least one processor in a video encoder, the method comprising:
receiving a bitstream that includes an image;
partitioning the image into a plurality of blocks;
compressing each of the plurality of blocks by a neural network-based encoder;
obtaining compressed features by obtaining a compressed feature for each block from among the plurality of blocks in the image;
processing the compressed features by an entropy encoder to generate a first compressed bitstream;
obtaining a plurality of reshaped compressed features by concatenating the compressed features such that a number of reshaped compressed features is less than a number of compressed features, wherein the concatenating the compressed features comprises concatenating the compressed feature blocks into reshaped compressed feature blocks that are larger than the compressed feature blocks, wherein a first reshaped compressed feature block has a square shape and a second reshaped feature block has a rectangular shape, wherein at least two of the reshaped compressed feature blocks are a different size from each other;
processing the plurality of reshaped compressed features by the entropy encoder to generate a second compressed bitstream; and
encoding the bitstream including the image based on the second compressed bitstream.