drprojects / DeepViewAgg

[CVPR'22 Best Paper Finalist] Official PyTorch implementation of the method presented in "Learning Multi-View Aggregation In the Wild for Large-Scale 3D Semantic Segmentation"
Other
222 stars 24 forks source link

Issue when running the inference script on the test data #38

Closed adikrishnanucd closed 11 months ago

adikrishnanucd commented 12 months ago

Hello @drprojects,

I am getting this error when running the kitti360_inference.ipynb on the test dataset:

ValueError: UnimodalBranch.out_channels has not been set. Please set it to allow inference even when the modality has no data. 

I have narrowed this down to the ImageBatch of the data being as follows:

MMBatch(
    data = DataBatch(scattering=[591], norm=[591, 3], pos=[591, 3], grid_size=[8], origin_id=[591], mapping_index=[591], num_raw_points=[8], planarity=[591], linearity=[591], idx_window=[8], idx_center=[8], x=[591, 1], coords=[591, 3], batch=[591], ptr=[9])
    image = ImageBatch(num_settings=1, num_views=0, num_points=591, device=cpu)
)

I need clarification on why the ImageBatch has no data even though the same transforms are used on the validation dataset, which works properly for running inference. Please point me to a few possibilities as to why I am getting this error.