コード例 #1
0
def create_lmdb_meta_file(num_train_examples, num_val_examples, num_test_examples, path_to_lmdb_meta_file):
    print('Saving meta file to %s...' % path_to_lmdb_meta_file)
    meta = Meta()
    meta.num_train_examples = num_train_examples
    meta.num_val_examples = num_val_examples
    meta.num_test_examples = num_test_examples
    meta.save(path_to_lmdb_meta_file)
コード例 #2
0
def create_tfrecords_meta_file(num_train_examples,num_val_examples, num_test_examples,path_to_tfrecords_meta_file):
    print 'Saving meta file to %s...' % path_to_tfrecords_meta_file
    meta = Meta()
    meta.num_train_examples = num_train_examples
    meta.num_val_examples = num_val_examples
    meta.num_test_examples = num_test_examples
    meta.save(path_to_tfrecords_meta_file)
コード例 #3
0
def create_tfrecords_meta_file(num_train_examples, num_val_examples, num_test_examples,
                               path_to_tfrecords_meta_file):
    print 'Saving meta file to %s...' % path_to_tfrecords_meta_file
    meta = Meta()
    meta.num_train_examples = num_train_examples
    meta.num_val_examples = num_val_examples
    meta.num_test_examples = num_test_examples
    meta.save(path_to_tfrecords_meta_file)
コード例 #4
0
 def create_tfrecords_meta_file(self,
                                dict_sample_number,
                                meta_name='batches_meta.json'):
     print('Saving meta file to %s...' % self.input_dir)
     meta = Meta(**dict_sample_number)
     with open(self.input_dir + '/cifar-10-batches-py/batches.meta',
               'rb') as f:
         content = pickle.load(f, encoding='latin1')
         meta.categories = content['label_names']
     meta.save(self.input_dir + '/' + meta_name)
コード例 #5
0
def create_to_tfrecords_meta_file(num_test_examples,
                                  path_to_tfrecords_meta_file):
    print('Saving meta file to %s...' % (path_to_tfrecords_meta_file))
    meta = Meta()
    meta.num_test_examples = num_test_examples
    meta.save(path_to_tfrecords_meta_file)