def get_data(self, args, path):
traj_locations = get_traj_locations(path,
criterion=shot_length_criterion)
transform_ts_data = transform_producer_ts(1)
bball_dataset = BballDataset(traj_locations,
transform=transform_ts_data)
return bball_dataset
def get_data(self, args, path):
traj_locations = get_traj_locations(path,
criterion=shot_length_criterion)
print(path, len(traj_locations))
transform_flat_data = transform_producer_flat(1, crop_len=args.trajlen)
bball_dataset = BballDataset(traj_locations,
transform=transform_flat_data)
return bball_dataset