search

yangcaoai / CoDA_NeurIPS2023

Official code for NeurIPS2023 paper: CoDA: Collaborative Novel Box Discovery and Cross-modal Alignment for Open-vocabulary 3D Object Detection
https://yangcaoai.github.io/publications/CoDA.html
MIT License
175 stars 15 forks source link

The train errors AttributeError: 'Model3DETRPredictedBoxDistillationHead' object has no attribute 'if_select_box_by_objectness' #13

Closed ZZ2490 closed 3 weeks ago

ZZ2490 commented 2 months ago

I encountered the following problem when running bash scripts/coda_sunrgbd_stage1.sh image When reviewing the code, I found that during initialization, the default setting is if_select_box_by_objectness=False. Do I need to modify the code?

ZZ2490 commented 2 months ago

I modified the code by setting self.if_select_box_by_objectness = False and self.if_keep_box = False. This allows the code to run without errors. Is it okay to make these modifications, and will it affect the final results? image

yangcaoai commented 2 months ago

Hi, they should be False. I have fixed the bug in https://github.com/yangcaoai/CoDA_NeurIPS2023/blob/main/models/model_3detr.py#L192 Please have a check

ZZ2490 commented 2 months ago

Thank you very much, I'll give it a try.

yangcaoai commented 2 months ago

You're welcome. Feel free to ask if you have any questions. If you found the repo helpful, please consider giving it a star :)