niftynet.layer.loss_gan module

class niftynet.layer.loss_gan.LossFunction(loss_type='CrossEntropy', loss_func_params={}, name='loss_function')

Bases: niftynet.layer.base_layer.Layer

layer_op(pred_real, pred_fake, var_scope=None)

make_callable_loss_func(type_str)

niftynet.layer.loss_gan.cross_entropy_function(is_real, softness=0.1)