邱浩楠 commited on
Commit
9cfe57b
1 Parent(s): 3238866
models/sample_model.py CHANGED
@@ -52,7 +52,7 @@ class SampleModel(BaseModel):
52
  By default, we use vanilla GAN loss, UNet with batchnorm, and aligned datasets.
53
  """
54
  # changing the default values to match the pix2pix paper (https://phillipi.github.io/pix2pix/)
55
- parser.set_defaults(norm='batch', netG='unet_256', dataset_mode='noise', num_test = 32, model='sample')
56
  parser.add_argument('--pose_path', type=str, default='', help='path for pose net')
57
  parser.add_argument('--rnn_path', type=str, default='', help='path for rnn net')
58
  parser.add_argument('--n_frames_G', type=int, default=60)
 
52
  By default, we use vanilla GAN loss, UNet with batchnorm, and aligned datasets.
53
  """
54
  # changing the default values to match the pix2pix paper (https://phillipi.github.io/pix2pix/)
55
+ parser.set_defaults(norm='batch', netG='unet_256', dataset_mode='noise', num_test = 32)
56
  parser.add_argument('--pose_path', type=str, default='', help='path for pose net')
57
  parser.add_argument('--rnn_path', type=str, default='', help='path for rnn net')
58
  parser.add_argument('--n_frames_G', type=int, default=60)
options/test_options.py CHANGED
@@ -16,7 +16,7 @@ class TestOptions(BaseOptions):
16
  parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
17
  parser.add_argument('--num_test', type=int, default=50, help='how many test images to run')
18
  # rewrite devalue values
19
- parser.set_defaults(model='test')
20
  # To avoid cropping, the load_size should be the same as crop_size
21
  parser.set_defaults(load_size=parser.get_default('crop_size'))
22
  self.isTrain = False
 
16
  parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
17
  parser.add_argument('--num_test', type=int, default=50, help='how many test images to run')
18
  # rewrite devalue values
19
+ # parser.set_defaults(model='test')
20
  # To avoid cropping, the load_size should be the same as crop_size
21
  parser.set_defaults(load_size=parser.get_default('crop_size'))
22
  self.isTrain = False