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. | |