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'
Loading...
Searching...
No Matches
Classes | Namespaces | Functions
custom_dataset.py File Reference

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.