MSE-CNN Implementation 1
Code database with the implementation of MSE-CNN, from the paper 'DeepQTMT: A Deep Learning Approach for Fast QTMT-based CU Partition of Intra-mode VVC'
|
▼Nmsecnn_raulkviana | |
▼Ncustom_dataset | |
CCUDatasetBase | Dataset stage oriented with capability of loading different files and it's supposed to be used with the function dataset_utils.change_labels_function_again |
CCUDatasetStg2Compl | |
CCUDatasetStg3Compl | |
CCUDatasetStg4Compl | Dataset stage oriented with capability of loading different files and it's supposed to be used with the function dataset_utils.change_labels_function_again |
CCUDatasetStg5Compl | Dataset stage oriented with capability of loading different files and it's supposed to be used with the function dataset_utils.change_labels_function_again |
CCUDatasetStg6Compl | Dataset stage oriented with capability of loading different files and it's supposed to be used with the function dataset_utils.change_labels_function_again |
CSamplerStg4 | |
CSamplerStg5 | |
CSamplerStg6 | |
▼Ndataset_utils | |
CVideoCaptureYUV | |
CVideoCaptureYUVV2 | |
▼Nmsecnn | |
CLossFunctionMSE | |
CLossFunctionMSE_Ratios | |
CMseCnnStg1 | |
CMseCnnStgX | |
CQP_half_mask |