← back to Handbag Authentication

handbag_data/github_datasets/iGAN/train_dcgan/get_num_images.py

10 lines

import argparse
import load


parser = argparse.ArgumentParser(description='get #images in a hdf5 dataset.')
parser.add_argument('--data_file', dest='data_file', help='the location of dataset file', default='../datasets/outdoor_64.hdf5', type=str)
args = parser.parse_args()
_, _, _, _, ntrain, ntest = load.load_imgs(ntrain=None, ntest=None, batch_size=128, data_file=args.data_file)
print('dataset: %s; #training images: %d; #test images: %d' % (args.data_file, ntrain, ntest))