search

samxuxiang / SkexGen

[ICML 2022] Official PyTorch Implementation of "SkexGen: Autoregressive Generation of CAD Construction Sequences with Disentangled Codebooks".
https://samxuxiang.github.io/skexgen
MIT License
87 stars 17 forks source link

Add note about downloading train_val_test_split.json #2

Closed JoeLambourne closed 2 years ago

JoeLambourne commented 2 years ago

Why?

The script parse.py exists with

Loading obj data...
Traceback (most recent call last):
  File "/home/lambouj/projects/SkexGen/utils/parse.py", line 22, in <module>
    train_samples, test_samples, val_samples = parser.load_all_obj()
  File "/home/lambouj/projects/SkexGen/utils/dataset.py", line 22, in load_all_obj
    with open('../data/train_val_test_split.json') as f:
FileNotFoundError: [Errno 2] No such file or directory: '../data/train_val_test_split.json'

What?

I add a note to the README to tell a user that they also need to download the deepcad train_val_test_split.json file.

I also add the data folder and a .gitkeep file to force git to create that folder when the repo is cloned. Now it's super clear where to put the data downloaded from DeepCAD.