isl-org / Open3D-ML

An extension of Open3D to address 3D Machine Learning tasks
Other
1.88k stars 321 forks source link

AttributeError: 'ConfigDict' object has no attribute 'augment' #594

Closed raviS123-dot closed 7 months ago

raviS123-dot commented 1 year ago

import open3d.ml.torch as ml3d from open3d.ml.torch.models import PointPillars from open3d.ml.torch.pipelines import SemanticSegmentation

dataset = ml3d.datasets.KITTI(dataset_path='/home/v3/Open3D-ML-master/ml3d/datasets/dataset', use_cache=True) model = PointPillars() pipeline = ObjectDetection(model=model, dataset=dataset, max_epoch=100) pipeline.run_train()


Traceback (most recent call last): File "/usr/lib/python3.10/idlelib/run.py", line 578, in runcode exec(code, self.locals) File "/home/v3/Desktop/kitti/train1.py", line 8, in model = PointPillars() File "/home/v3/.local/lib/python3.10/site-packages/open3d/_ml3d/torch/models/point_pillars.py", line 84, in init self.augmenter = ObjdetAugmentation(self.cfg.augment, seed=self.rng) File "/home/v3/.local/lib/python3.10/site-packages/open3d/_ml3d/utils/config.py", line 244, in getattr return getattr(self._cfg_dict, name) File "/home/v3/.local/lib/python3.10/site-packages/open3d/_ml3d/utils/config.py", line 27, in getattr raise ex AttributeError: 'ConfigDict' object has no attribute 'augment'

samuelsaqueiroz commented 9 months ago

I have the same problem here. Do you have any updates? Did you solved it?

ricardosanunes commented 8 months ago

Same problem here. Were you able to solve it @raviS123-dot ?

willlauer commented 8 months ago

Seems the complete config can be loaded from a corresponding yaml file https://github.com/isl-org/Open3D-ML/blob/fcf97c07bf7a113a47d0fcf63760b245c2a2784e/ml3d/configs/pointpillars_lyft.yml, or passed in manually e.g.


model = PointPillars(
    voxelize={
        'voxel_size': [0.2, 0.2, 0.2]
    },
    augment={})
raviS123-dot commented 7 months ago

thanks @willlauer