Config Release of Multiple Classes

LittlePey / SFD

Sparse Fuse Dense: Towards High Quality 3D Detection with Depth Completion (CVPR 2022, Oral)

Apache License 2.0

262 stars 35 forks source link

Config Release of Multiple Classes #24

Open Son-Goku-gpu opened 1 year ago

Son-Goku-gpu commented 1 year ago

Great work! may I know if you can release the config of multi-class training?

LittlePey commented 1 year ago

Hi, sorry for the late reply, you can modify our yaml file to support 3 classes according to pv_rcnn.yaml. We did not do special design for ped and cyc when we train our 3 classes model.

Gmonster-24 commented 1 year ago

@LittlePey 您好，如果我要训练多目标检测的话，是不是不需要修改生成pseudo_point信息那部分代码，pseudo_point中已经有pedestrian和cyclist的标注信息和ground truth了

LittlePey commented 1 year ago

是的，您不需要修改

Gmonster-24 commented 1 year ago

是的，您不需要修改

非常感谢！