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()
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)