facebookresearch / co3d

Tooling for the Common Objects In 3D dataset.
Other
921 stars 71 forks source link

Fixed dataset error when not loading masks #44

Open jasonyzhang opened 2 years ago

jasonyzhang commented 2 years ago

When masks are not loaded by the Co3dDataset, fg_probability, full_path, and bbox_xywh are not set, resulting in a NameError. Fixed this by setting a default value.