search

manycore-research / PlankAssembly

[ICCV 2023] PlankAssembly: Robust 3D Reconstruction from Three Orthographic Views with Learnt Shape Programs
https://manycore-research.github.io/PlankAssembly/
GNU Affero General Public License v3.0
61 stars 6 forks source link

Data format #4

Closed guist closed 10 months ago

guist commented 10 months ago

Hello,

I have an additional question regarding the way of preparing the data for inference.

When I run the testing script using the configs/train.yaml, I get the following error:

File "/home/guist/PlankAssembly-atlas/plankassembly/data.py", line 246, in getitem svgs = info['svgs'] KeyError: 'svgs'

I checked the json file and there is indeed no 'svgs' field so I was wondering how to modify the file so that it works. More generally, how should I prepare my custom images so that I can apply your model on them?

Thanks a lot for the answer.

Best regards, Guillaume

bertjiazheng commented 10 months ago

Hi,

The PlankAssembly dataset provides shape programs for the 3D model. To train or test the model, you should first prepare the data following here.

The input data should include three orthographic line drawings. Refer to the line drawing renderings scripts for more information.

Best, Jia

guist commented 10 months ago

Hi,

Thanks for your explanations, I could prepare the data and run the full testing script.

Best, Guillaume