cminst/ModelNet40

import torch_geometric
from torch_geometric.datasets import ModelNet
from torch_geometric.transforms import Compose, SamplePoints, KNNGraph

torch_geometric.seed.seed_everything(42)

# Standard setup
pre_transform = Compose([
    SamplePoints(1024),
    KNNGraph(k=16)
])

path = '/mnt/lrmc_pool/ModelNet40'
train_dataset = ModelNet(path, '40', train=True, transform=pre_transform)
test_dataset = ModelNet(path, '40', train=False, transform=pre_transform)



%cd /mnt/lrmc_pool
!tar -I pigz -cf ModelNet40_s1024_k16.tar.gz ModelNet40


from huggingface_hub import upload_folder, upload_file, create_repo

try:
    create_repo("cminst/ModelNet40", repo_type="dataset")
    print("Dataset upload completed!")
except Exception as e:
    print(f"Repository might already exist or upload failed: {e}")

upload_file(
    path_or_fileobj="/mnt/lrmc_pool/ModelNet40_s1024_k16.tar.gz",
    repo_id="cminst/ModelNet40",
    path_in_repo="ModelNet40_s1024_k16.tar.gz",
    repo_type="dataset",
)