def define_D(which_netD):
    if which_netD == 'FFC':
        return networks.FFC()
    elif which_netD == 'NoBNMultPathdilationNet':
        return networks.NoBNMultPathdilationNet()
    elif which_netD == 'SinglePathdilationSingleOutputNet':
        return networks.SinglePathdilationSingleOutputNet()
    elif which_netD == 'SinglePathdilationMultOutputNet':
        return networks.SinglePathdilationMultOutputNet()
    elif which_netD == 'NoBNSinglePathdilationMultOutputNet':
        return networks.NoBNSinglePathdilationMultOutputNet()
    elif which_netD == 'RandomMultPathdilationNet':
        return networks.RandomMultPathdilationNet()
    elif which_netD == 'lsgan_D':
        return networks.lsgan_D()
    elif which_netD == 'lsganMultOutput_D':
        return networks.lsganMultOutput_D()
Exemple #2
0
def define_D(which_netD, input_nc):
    if which_netD == 'NoBNSinglePathdilationMultOutputNet':
        return networks.NoBNSinglePathdilationMultOutputNet(input_nc)
    elif which_netD == 'lsganMultOutput_D':
        return networks.lsganMultOutput_D(input_nc)