parser = argparse.ArgumentParser()

parser.add_argument("--cache_dir", type=str, default=None)

parser.add_argument("batch", type=str)

args = parser.parse_args()

print("creating classifier...")

clf = PointNetClassifier(devices=get_torch_devices(), cache_dir=args.cache_dir)

print("computing batch predictions")

_, preds = clf.features_and_preds(NpzStreamer(args.batch))