A recursive implementation of VVC partition from [1].
pip3 install -r requirements.txt
python3 src/main.py <input_image> <output_image>
Don't use input_image
with resolution higher than 1080p, or the runtime would be extremely long.
CTU_SIZE
andMAX_DEPTH
insrc/main.py
specify the size of coding tree unit and the maximum partition depth.MIN_CU_SIZE
insrc/CodingUnit.py
specifies the minimum coding unit size.- In
src/early_terminate.py
, if a coding unit's variance is lower thanTH_1
, further partition will be terminated. - In
src/gradient_select.py
,TH_2
andTH_3
specify the threshold for selecting QT over MT.
Note that different parameters leads to different time required for execution.
[1] Y. Fan, J. Chen, H. Sun, J. Katto and M. Jing, "A Fast QTMT Partition Decision Strategy for VVC Intra Prediction," in IEEE Access, vol. 8, pp. 107900-107911, 2020