search

lukemelas / projection-conditioned-point-cloud-diffusion

Official code for "Projection-Conditioned Point Cloud Diffusion for Single-Image 3D Reconstruction" (CVPR 2023)
177 stars 15 forks source link

ShapeNet checkpoint & test scripts #2

Open mingrui-zhao opened 1 year ago

mingrui-zhao commented 1 year ago

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 commented 1 year ago

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

xXuHaiyang commented 1 year ago

Also interested

Estellepark commented 1 year ago

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 commented 7 months ago

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.