yifanlu0227 / HEAL

[ICLR2024] HEAL: An Extensible Framework for Open Heterogeneous Collaborative Perception ➡️ All You Need for Multi-Modality Collaborative Perception!
Other
147 stars 9 forks source link

There is a bug in heter_model_late.py. #11

Closed xhjy2020 closed 5 months ago

xhjy2020 commented 5 months ago

Hello! In opencood/models/heter_model_late.py , line 14, from opencood.models.sub_modules.roi_align_fe import RoIAlignFeatureExtractor, there is no 'roi_align_fe' files.

I want to train no-fusion models. I notice that there are some xxx_pretrain.yaml, are these files for train late-fusion models? No-fusion is only supported during inference? image

yifanlu0227 commented 5 months ago

Hi! RoIAlignFeatureExtractor is no longer needed. I remove it in the latest commit.

The late fusion and no fusion share the same yaml. They are both trained on a single agent, without collaboration. In the inference stage, use

xhjy2020 commented 5 months ago

I got it, thanks for your patient elaboration~