|
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'
|
Classes | |
| class | msecnn_raulkviana.custom_dataset.CUDatasetBase |
| 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. More... | |
| class | msecnn_raulkviana.custom_dataset.CUDatasetStg5Compl |
| 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. More... | |
| class | msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl |
| class | msecnn_raulkviana.custom_dataset.CUDatasetStg6Compl |
| 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. More... | |
| class | msecnn_raulkviana.custom_dataset.CUDatasetStg4Compl |
| 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. More... | |
| class | msecnn_raulkviana.custom_dataset.CUDatasetStg3Compl |
| class | msecnn_raulkviana.custom_dataset.SamplerStg6 |
| class | msecnn_raulkviana.custom_dataset.SamplerStg5 |
| class | msecnn_raulkviana.custom_dataset.SamplerStg4 |
Namespaces | |
| namespace | msecnn_raulkviana |
| namespace | msecnn_raulkviana.custom_dataset |
Functions | |
| msecnn_raulkviana.custom_dataset.yuv2bgr (matrix) | |
| Converts yuv matrix to bgr matrix. | |
| msecnn_raulkviana.custom_dataset.bgr2yuv (matrix) | |
| Converts BGR matrix to YUV matrix. | |
| msecnn_raulkviana.custom_dataset.get_cu (f_path, f_size, cu_pos, cu_size, frame_number) | |
| Get CU from image. | |
| msecnn_raulkviana.custom_dataset.get_cu_v2 (img, f_size, cu_pos, cu_size) | |
| Get CU from image. | |
| msecnn_raulkviana.custom_dataset.get_file_size (name) | |
| Retrieves information about the YUV file info (width and height) | |
| msecnn_raulkviana.custom_dataset.resize (img, scale_percent) | |
| Resizes a BGR image. | |
| msecnn_raulkviana.custom_dataset.show_CU (image, cu_size, cu_pos) | |
| Shows CU in a image. | |
| msecnn_raulkviana.custom_dataset.show_all_CUs (CUDataset, file_name, POC, cu_size) | |
| Shows all CUs in a image with a specific size, TO BE USED WITH THE CU ORIENTE LABELS. | |