File size: 396 Bytes
9913017 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import argparse
from datasets.flow_datasets import AnimeRun, fetch_dataloader
if __name__ == '__main__':
data = AnimeRun(split='test', dstype='Frame_Anime')
print(len(data))
args = argparse.Namespace()
args.stage = 'anime'
args.dstype = 'Frame_Anime'
args.image_size = [384, 768]
args.batch_size = 12
loader = fetch_dataloader(args)
print(len(loader))
|