File size: 605 Bytes
0d52562 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | #### This script is used to store the pickle files for the ETH dataset from the txt files.
from data.dataloader_eth_ucy import ETHDatasetSocialGAN
import argparse
def parse_args():
argparser = argparse.ArgumentParser()
argparser.add_argument('--data_dir', type=str, required=True, help='Directory where the data is stored.')
return argparser.parse_args()
if __name__ == '__main__':
args = parse_args()
for subset in ['eth', 'hotel', 'univ', 'zara1', 'zara2']:
dataset = ETHDatasetSocialGAN(args.data_dir + f'/{subset}', subset=subset)
print("Done storing pickle files.") |