Transform-based image coding method, and device therefor
An image decoding method according to the present document may comprise the steps of: if a current block is partitioned into sub-partition transform blocks, deriving a prediction sample of the current block on the basis of intra prediction mode information; determining an LFNST set including LFNST matrices on the basis of an intra prediction mode derived from the intra prediction mode information; selecting one of the LFNST matrices on the basis of the LFNST set and the LFNST index; deriving transform coefficients for the sub-partition transform blocks on the basis of the selected LFNST matrix; and deriving residual samples for the current block on the basis of the transform coefficients.
1 . An image decoding apparatus, comprising:
a memory; and
at least one processor connected to the memory, the at least one processor configured to:
obtain intra prediction mode information and low-frequency non-separable transform (LFNST) index information for a current block from a bitstream,
wherein the current block is divided into a plurality of sub-partition transform blocks based on an intra sub-partition mode in which intra prediction is performed on each of the sub-partition transform blocks in the current block, and
wherein the LFNST index information for the current block is obtained, from the bitstream, based on a width and a height of a sub-partition transform block resulting from dividing the current block being greater than or equal to 4,
derive prediction samples of the sub-partition transform block based on an intra prediction mode derived from the intra prediction mode information for the current block,
determine an LFNST set including LFNST matrices based on the intra prediction mode,
select one of the LFNST matrices based on the LFNST set and the LFNST index information,
derive transform coefficients for the sub-partition transform block based on the one of the LFNST matrices,
derive residual samples for the sub-partition transform block based on the transform coefficients, and
generate reconstructed samples for the sub-partition transform block based on the prediction samples and the residual samples.
2 . The image decoding apparatus of claim 1 , wherein the LFNST set and the LFNST index information are applied to the sub-partition transform blocks divided from the current block.
3 . The image decoding apparatus of claim 1 , wherein the intra prediction mode is applied to the sub-partition transform blocks divided from the current block.
4 . The image decoding apparatus of claim 1 , wherein the transform coefficients are derived using the one of the LFNST matrices based on the width and the height of the sub-partition transform block being greater than or equal to 4.
5 . The image decoding apparatus of claim 1 , wherein the current block is vertically divided based on a size (width×height) of the current block being 8×4, and
wherein the current block is horizontally divided based on the size (width×height) of the current block being 4×8.
6 . The image decoding apparatus of claim 1 , wherein the current block is divided into four sub-partition transform blocks in a horizontal or vertical direction based on a size (width×height) of the current block being greater than 4×8 or 8×4.
7 . An image encoding apparatus, comprising:
a memory; and
at least one processor connected to the memory, the at least one processor configured to:
derive, based on an intra prediction mode of a current block, prediction samples for a sub-partition transform block resulting from dividing the current block,
wherein the current block is divided into a plurality of sub-partition transform blocks based on an intra sub-partition mode in which intra prediction is performed on each of the sub-partition transform blocks in the current block,
derive residual samples for the sub-partition transform block based on the prediction samples,
apply a primary transform to the residual samples to derive transform coefficients,
derive modified transform coefficients for the sub-partition transform block based on a low-frequency non-separable transform (LFNST) set mapped to the intra prediction mode and an LFNST matrix included in the LFNST set, and
encode image information including LFNST index information related to the LFNST matrix and intra prediction mode information related to the intra prediction mode,
wherein the LFNST index information is encoded into a bitstream based on a width and a height of the sub-partition transform block being greater than or equal to 4.
8 . The image encoding apparatus of claim 7 , wherein the LFNST set and the LFNST matrix are applied to the sub-partition transform blocks.
9 . The image encoding apparatus of claim 7 , wherein the modified transform coefficients are derived using the LFNST matrix based on the width and the height of the sub-partition transform block being greater than or equal to 4.
10 . The image encoding apparatus of claim 7 , wherein the current block is vertically divided based on a size (width×height) of the current block being 8×4, and
wherein the current block is horizontally divided based on the size (width×height) of the current block being 4×8.
11 . The image encoding apparatus of claim 7 , wherein the current block is divided into four sub-partition transform blocks in a horizontal or vertical direction based on a size (width×height) of the current block being greater than 4×8 or 8×4.
12 . A transmission apparatus of data for an image, comprising:
at least one processor configured to obtain a bitstream for the image, wherein the bitstream is generated based on deriving, based on an intra prediction mode of a current block, prediction samples for a sub-partition transform block resulting from dividing the current block, the current block being divided into a plurality of sub-partition transform blocks based on an intra sub-partition mode in which intra prediction is performed on each of the sub-partition transform blocks in the current block, deriving residual samples for the sub-partition transform block based on the prediction samples, applying a primary transform to the residual samples to derive transform coefficients, deriving modified transform coefficients for the sub-partition transform block based on a low-frequency non-separable transform (LFNST) set mapped to the intra prediction mode and an LFNST matrix included in the LFNST set, and encoding image information including LFNST index information related to the LFNST matrix and intra prediction mode information related to the intra prediction mode; and
a transmitter configured to transmit the data including the bitstream,
wherein the LFNST index information is encoded into the bitstream based on a width and a height of the sub-partition transform block being greater than or equal to 4.