None type in preprocess_data.py

nv-tlabs / ATISS

Code for "ATISS: Autoregressive Transformers for Indoor Scene Synthesis", NeurIPS 2021

Other

258 stars 55 forks source link

None type in preprocess_data.py #6

Closed wamiq-reyaz closed 2 years ago

wamiq-reyaz commented 2 years ago

Hello,

For some models, the category in the json is None, and preprocess_data.py does not handle that well, for example I get.

{'model_id': '00e4b7dd-08a6-4433-a439-856e4b5de58a', 'super-category': 'Others', 'category': None, 'style': 'Minimalist', 'material': 'Composition', 'theme': 'Gold Foil'}

Should this be handled by the blacklists? Or am I doing something wrong?

paschalidoud commented 2 years ago

Hi @wamiq-reyaz,

I am very sorry for my super late reply! @nazcaspider suggested a fix that should fix this issue. It seems that 3D-FRONT was updated and now contains some categories that are defined as None. I just committed the fix and hopefully the issue will be resolved.

I am closing the issue for now but please feel free to reopen it in case the issue is not solved.

Cheers, Despoina