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
msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios Member List

This is the complete list of members for msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios, including all inherited members.

__init__(self, pm, use_mod_cross_entropy=True, beta=1, alpha=0.5)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
alphamsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
betamsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
forward(self, pred, labels, RD)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
get_min_RDs(self, RDs)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
get_proportion_CUs(self, labels)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
last_lossmsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
last_predmsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
last_RDmsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
MAX_LOSSmsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
MAX_RDmsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
pmmsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
remove_inf_values(self, tensor)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
remove_values_above(self, RDs, max_val)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
remove_values_lower(self, tensor, max_val, subst_val)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
remove_zero(self, RDs)msecnn_raulkviana.msecnn.LossFunctionMSE_Ratios
use_mod_cross_entropymsecnn_raulkviana.msecnn.LossFunctionMSE_Ratios