ShapeNet checkpoint & test scripts

[mingrui-zhao]

Hi,

The current sampling codes look contained to the CO3D dataset. Would it be possible to also release the checkpoint and test scripts on the ShapeNet dataset?

Thanks in advance!

[xiexh20]

I am also very interested in the shapenet pretrained models. Thanks!

[xXuHaiyang]

Also interested

[Estellepark]

Hello everyone, has anyone successfully loaded the ShapeNet dataset for training? I tried it but it failed during the data loading phase. I would be very grateful if you can provide valuable suggestions.

[xiexh20]

For people who are interested, I implemented my own dataloader for shapnet here: https://github.com/xiexh20/HDM/blob/main/dataset/r2n2_dataset.py

It is compatible with the pc2 framework, you just need to change the dataset in __init__.py file: https://github.com/xiexh20/HDM/blob/main/dataset/__init__.py#L105

I haven't reproduced the exact reconstruction numbers from the paper though.