How to change multi head to single?

V2AI / Det3D

World's first general purpose 3D object detection codebse.

https://arxiv.org/abs/1908.09492

Apache License 2.0

1.48k stars 299 forks source link

How to change multi head to single? #137

Closed tojimahammatov closed 3 years ago

tojimahammatov commented 3 years ago

How to change a config file for single head?

Hi, thanks for great work.

I saw there are three heads available in bbox_heads directory, and I want to train single head with pointpillar (nuscenes). Obviously bbox_head should be replaced in the config, but how about other parameters (tasks, multi_class_nms etc) ?

poodarchu commented 3 years ago

you can chang tasks in config to one.