hylee commited on
Commit
2ea7ce8
·
1 Parent(s): 06aeeda
Files changed (2) hide show
  1. APDrawingGAN2/options/test_options.py +27 -0
  2. app.py +0 -13
APDrawingGAN2/options/test_options.py CHANGED
@@ -4,6 +4,21 @@ from .base_options import BaseOptions
4
  class TestOptions(BaseOptions):
5
  def initialize(self, parser):
6
  parser = BaseOptions.initialize(self, parser)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  parser.add_argument('--ntest', type=int, default=float("inf"), help='# of test examples.')
8
  parser.add_argument('--results_dir', type=str, default='./results/', help='saves results here.')
9
  parser.add_argument('--aspect_ratio', type=float, default=1.0, help='aspect ratio of result images')
@@ -17,6 +32,18 @@ class TestOptions(BaseOptions):
17
  parser.add_argument('--pretrain', action='store_true', help='pretrain stage, no dt loss, no ae')
18
 
19
  parser.set_defaults(model='test')
 
 
 
 
 
 
 
 
 
 
 
 
20
  # To avoid cropping, the loadSize should be the same as fineSize
21
  parser.set_defaults(loadSize=parser.get_default('fineSize'))
22
  self.isTrain = False
 
4
  class TestOptions(BaseOptions):
5
  def initialize(self, parser):
6
  parser = BaseOptions.initialize(self, parser)
7
+
8
+ # '''
9
+ # python test.py --dataroot dataset/test_single --name apdrawinggan++_author --model test --use_resnet --netG resnet_9blocks --which_epoch 150 --how_many 1000 --gpu_ids 0 --gpu_ids_p 0 --imagefolder images-single
10
+ # '''
11
+ # opt.dataroot = 'dataset/test_single'
12
+ # opt.name = 'apdrawinggan++_author'
13
+ # opt.model = 'test'
14
+ # opt.use_resnet = True
15
+ # opt.netG = 'resnet_9blocks'
16
+ # opt.which_epoch = 150
17
+ # opt.how_many = 1000
18
+ # opt.gpu_ids = ''
19
+ # opt.gpu_ids_p = ''
20
+ # opt.imagefolder = 'images-single'
21
+
22
  parser.add_argument('--ntest', type=int, default=float("inf"), help='# of test examples.')
23
  parser.add_argument('--results_dir', type=str, default='./results/', help='saves results here.')
24
  parser.add_argument('--aspect_ratio', type=float, default=1.0, help='aspect ratio of result images')
 
32
  parser.add_argument('--pretrain', action='store_true', help='pretrain stage, no dt loss, no ae')
33
 
34
  parser.set_defaults(model='test')
35
+
36
+ parser.set_defaults(dataroot='dataset/test_single')
37
+ parser.set_defaults(name='apdrawinggan++_author')
38
+ parser.set_defaults(use_resnet=True)
39
+ parser.set_defaults(netG='resnet_9blocks')
40
+ parser.set_defaults(which_epoch=150)
41
+ parser.set_defaults(how_many=1000)
42
+ parser.set_defaults(gpu_ids='-1')
43
+ parser.set_defaults(gpu_ids_p='-1')
44
+ parser.set_defaults(imagefolder='images-single')
45
+
46
+
47
  # To avoid cropping, the loadSize should be the same as fineSize
48
  parser.set_defaults(loadSize=parser.get_default('fineSize'))
49
  self.isTrain = False
app.py CHANGED
@@ -184,19 +184,6 @@ def main():
184
  opt.no_flip = True # no flip
185
  opt.display_id = -1 # no visdom display
186
 
187
- '''
188
- python test.py --dataroot dataset/test_single --name apdrawinggan++_author --model test --use_resnet --netG resnet_9blocks --which_epoch 150 --how_many 1000 --gpu_ids 0 --gpu_ids_p 0 --imagefolder images-single
189
- '''
190
- opt.dataroot = 'dataset/test_single'
191
- opt.name = 'apdrawinggan++_author'
192
- opt.model = 'test'
193
- opt.use_resnet = True
194
- opt.netG = 'resnet_9blocks'
195
- opt.which_epoch = 150
196
- opt.how_many = 1000
197
- opt.gpu_ids = ''
198
- opt.gpu_ids_p = ''
199
- opt.imagefolder = 'images-single'
200
 
201
  opt.checkpoints_dir = checkpoint_dir
202
 
 
184
  opt.no_flip = True # no flip
185
  opt.display_id = -1 # no visdom display
186
 
 
 
 
 
 
 
 
 
 
 
 
 
 
187
 
188
  opt.checkpoints_dir = checkpoint_dir
189