niftynet.layer.loss_gan module

class LossFunction(loss_type='CrossEntropy', loss_func_params=None, name='loss_function')

Bases: niftynet.layer.base_layer.Layer

make_callable_loss_func(type_str)

layer_op(pred_real, pred_fake, var_scope=None)

cross_entropy_function(is_real, softness=0.1)