コード例 #1
0
 def get_default_config(self):
     config = BasicModel.get_default_config(self)
     model_config = {
         'stride':
         1,
         'inception_v4_checkpoint_file':
         os.path.join(script_dir, '..', 'data', 'inception_v4.ckpt'),
         'batch_norm_decay':
         0.99,
         'batch_norm_epsilon':
         0.001,
         'output_size':
         29,
         'pad':
         32,
         'receptive_field_size':
         66,
         'projective_field_size':
         7,
         'contextual_pad':
         32,
         'normalize_inputs':
         False,
         'batch_size':
         64,
     }
     config.update(model_config)
     return config
コード例 #2
0
 def get_default_config(self):
     config = BasicModel.get_default_config(self)
     model_config = {
         'stride': 1,
         'img_rows_target': 289,
         'img_cols_target': 289,
         'normalize_inputs': True,
     }
     config.update(model_config)
     return config
コード例 #3
0
 def get_default_config(self):
     config = BasicModel.get_default_config(self)
     model_config = {
         'stride':
         1,
         'inception_v4_checkpoint_file':
         os.path.join(script_dir, '..', 'data', 'inception_v4.ckpt'),
         'batch_norm_decay':
         0.9,
         'batch_norm_epsilon':
         0.001,
         'output_size':
         29,  # Projective field size at Mixed_5b
         'pad':
         32,
         'receptive_field_size':
         2 * 33,
         'projective_field_size':
         7,
         'target_context_pad':
         23,  # ~(7./2.) / (128./33.)
         'target_embedding_pad':
         16,
         # The contextual pad should be less than 48, which is half of the
         # difference of the receptive fields of Mixed_5a and Mixed_5d, i.e. the
         # number of pixels that the additional receptive field can look ahead. It
         # should also not include so much black borders from data augmentation.
         'contextual_pad':
         32,
         # Excluding the additional pad needed in order to get the full receptive
         # field into view.
         'large_contextual_pad_unpadded':
         128,
         'large_contextual_pad':
         128 + 66,
         'normalize_inputs':
         False,
         'batch_size':
         4,
         'hidden_state_size':
         96,
         'draw_border':
         True,
         'loss_function':
         'l17',
     }
     config.update(model_config)
     return config