diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/cycle_gan_model.py | 9 | ||||
| -rw-r--r-- | models/models.py | 1 | ||||
| -rw-r--r-- | models/networks.py | 5 | ||||
| -rw-r--r-- | models/pix2pix_model.py | 2 |
4 files changed, 3 insertions, 14 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index b7b840d..bcc6a15 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -1,6 +1,4 @@ -import numpy as np import torch -import os from collections import OrderedDict from torch.autograd import Variable import itertools @@ -8,7 +6,6 @@ import util.util as util from util.image_pool import ImagePool from .base_model import BaseModel from . import networks -import sys class CycleGANModel(BaseModel): @@ -17,10 +14,6 @@ class CycleGANModel(BaseModel): def initialize(self, opt): BaseModel.initialize(self, opt) - - nb = opt.batchSize - size = opt.fineSize - # load/define networks # The naming conversion is different from those used in the paper # Code (paper): G_A (G), G_B (F), D_A (D_Y), D_B (D_X) @@ -200,7 +193,7 @@ class CycleGANModel(BaseModel): def get_current_errors(self): ret_errors = OrderedDict([('D_A', self.loss_D_A), ('G_A', self.loss_G_A), ('Cyc_A', self.loss_cycle_A), - ('D_B', self.loss_D_B), ('G_B', self.loss_G_B), ('Cyc_B', self.loss_cycle_B)]) + ('D_B', self.loss_D_B), ('G_B', self.loss_G_B), ('Cyc_B', self.loss_cycle_B)]) if self.opt.identity > 0.0: ret_errors['idt_A'] = self.loss_idt_A ret_errors['idt_B'] = self.loss_idt_B diff --git a/models/models.py b/models/models.py index d5bb9d8..39cc020 100644 --- a/models/models.py +++ b/models/models.py @@ -1,4 +1,3 @@ - def create_model(opt): model = None print(opt.model) diff --git a/models/networks.py b/models/networks.py index da2f59c..b118c6a 100644 --- a/models/networks.py +++ b/models/networks.py @@ -4,7 +4,6 @@ from torch.nn import init import functools from torch.autograd import Variable from torch.optim import lr_scheduler -import numpy as np ############################################################################### # Functions ############################################################################### @@ -434,6 +433,7 @@ class NLayerDiscriminator(nn.Module): else: return self.model(input) + class PixelDiscriminator(nn.Module): def __init__(self, input_nc, ndf=64, norm_layer=nn.BatchNorm2d, use_sigmoid=False, gpu_ids=[]): super(PixelDiscriminator, self).__init__() @@ -442,7 +442,7 @@ class PixelDiscriminator(nn.Module): use_bias = norm_layer.func == nn.InstanceNorm2d else: use_bias = norm_layer == nn.InstanceNorm2d - + self.net = [ nn.Conv2d(input_nc, ndf, kernel_size=1, stride=1, padding=0), nn.LeakyReLU(0.2, True), @@ -461,4 +461,3 @@ class PixelDiscriminator(nn.Module): return nn.parallel.data_parallel(self.net, input, self.gpu_ids) else: return self.net(input) - diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index 74a941e..9c46a19 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -1,6 +1,4 @@ -import numpy as np import torch -import os from collections import OrderedDict from torch.autograd import Variable import util.util as util |