search

THU-DA-6D-Pose-Group / GDR-Net

GDR-Net: Geometry-Guided Direct Regression Network for Monocular 6D Object Pose Estimation. (CVPR 2021)
https://github.com/THU-DA-6D-Pose-Group/GDR-Net
Apache License 2.0
274 stars 46 forks source link

How to train lm_pbr? #63

Closed cuge1995 closed 2 years ago

cuge1995 commented 2 years ago

when add lm_pbr in config, it fails

core/gdrn_modeling/dataset_factory.py", line 53, in register_datasets_in_cfg
    assert "DATA_CFG" in cfg and name in cfg.DATA_CFG, "no cfg.DATA_CFG.{}".format(name)
AssertionError: no cfg.DATA_CFG.lm_pbr
wangg12 commented 2 years ago

The datasets are registered at https://github.com/THU-DA-6D-Pose-Group/GDR-Net/blob/main/core/gdrn_modeling/datasets/lm_pbr.py#L296