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 Member List

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

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