search

DerrickXuNu / OpenCOOD

[ICRA 2022] An opensource framework for cooperative detection. Official implementation for OPV2V.
https://mobility-lab.seas.ucla.edu/opv2v/
Other
674 stars 100 forks source link

Intermediate fusion #153

Open cvips opened 4 days ago

cvips commented 4 days ago

Hi,

Why does the 'inference_intermediate_fusion(batch_data, model, dataset)' call the early fusion method?

`def inference_intermediate_fusion(batch_data, model, dataset): """ Model inference for early fusion.

Parameters
----------
batch_data : dict
model : opencood.object
dataset : opencood.EarlyFusionDataset

Returns
-------
pred_box_tensor : torch.Tensor
    The tensor of prediction bounding box after NMS.
gt_box_tensor : torch.Tensor
    The tensor of gt bounding box.
"""
return inference_early_fusion(batch_data, model, dataset)`