Python provide_data_from_image_files 예제들

예제 #1

 def test_provide_data_from_image_files_a_list_of_patterns(self):
     file_pattern = [os.path.join(self.testdata_dir, '*.jpg')]
     images = data_provider.provide_data_from_image_files(file_pattern,
                                                          batch_size=2,
                                                          shuffle=False,
                                                          patch_height=3,
                                                          patch_width=3,
                                                          colors=1)
     self.assertEqual(images.shape.as_list(), [2, 3, 3, 1])
     with self.test_session(use_gpu=True) as sess:
         sess.run(tf.local_variables_initializer())
         with tf.contrib.slim.queues.QueueRunners(sess):
             images_np = sess.run(images)
     self.assertEqual(images_np.shape, (2, 3, 3, 1))

예제 #2

파일: data_provider_test.py 프로젝트: ALISCIFP/models

 def test_provide_data_from_image_files_a_list_of_patterns(self):
   file_pattern = [os.path.join(self.testdata_dir, '*.jpg')]
   images = data_provider.provide_data_from_image_files(
       file_pattern,
       batch_size=2,
       shuffle=False,
       patch_height=3,
       patch_width=3,
       colors=1)
   self.assertEqual(images.shape.as_list(), [2, 3, 3, 1])
   with self.test_session(use_gpu=True) as sess:
     sess.run(tf.local_variables_initializer())
     with tf.contrib.slim.queues.QueueRunners(sess):
       images_np = sess.run(images)
   self.assertEqual(images_np.shape, (2, 3, 3, 1))

예제 #3

def _provide_real_images(batch_size, **kwargs):
    """Provides real images."""
    dataset_name = kwargs.get('dataset_name')
    dataset_file_pattern = kwargs.get('dataset_file_pattern')
    colors = kwargs['colors']
    final_height, final_width = train.make_resolution_schedule(
        **kwargs).final_resolutions
    if dataset_name is not None:
        return data_provider.provide_data(dataset_name=dataset_name,
                                          split_name='train',
                                          batch_size=batch_size,
                                          patch_height=final_height,
                                          patch_width=final_width,
                                          colors=colors)
    elif dataset_file_pattern is not None:
        return data_provider.provide_data_from_image_files(
            file_pattern=dataset_file_pattern,
            batch_size=batch_size,
            patch_height=final_height,
            patch_width=final_width,
            colors=colors)

예제 #4

파일: train_main.py 프로젝트: Exscotticus/models

def _provide_real_images(batch_size, **kwargs):
  """Provides real images."""
  dataset_name = kwargs.get('dataset_name')
  dataset_file_pattern = kwargs.get('dataset_file_pattern')
  colors = kwargs['colors']
  final_height, final_width = train.make_resolution_schedule(
      **kwargs).final_resolutions
  if dataset_name is not None:
    return data_provider.provide_data(
        dataset_name=dataset_name,
        split_name='train',
        batch_size=batch_size,
        patch_height=final_height,
        patch_width=final_width,
        colors=colors)
  elif dataset_file_pattern is not None:
    return data_provider.provide_data_from_image_files(
        file_pattern=dataset_file_pattern,
        batch_size=batch_size,
        patch_height=final_height,
        patch_width=final_width,
        colors=colors)