tusen-ai / SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).
Apache License 2.0
765 stars 96 forks source link

sample_groups parameter of DataBaseSampler #191

Closed ArseniuML closed 1 month ago

ArseniuML commented 2 months ago

As I can understand, this parameter is related to sampling frequency of objects of the class. If this parameter = 0, objects of this class would not be sampled.

In argo2-3d-26class.py you define sample_groups as follows:

sample_group_1 = {k:1 for k in group1}
sample_group_2 = {k:2 for k in group2}
sample_group_3 = {k:2 for k in group3}
sample_group_4 = {k:1 for k in group4}
sample_group_5 = {k:2 for k in group5}
sample_group_6 = {k:2 for k in group6}
#merge all groups
sample_groups = {**sample_group_1, **sample_group_2, **sample_group_3, **sample_group_4, **sample_group_5, **sample_group_6}
sample_groups.update({'Wheelchair':0, 'Dog':0, 'Message_board_trailer':0})

How did you obtain these numbers(0, 1, or 2)?

In the FSDv2 paper you write about FSD:

the clustering part in instance segmentation voted centers necessitates pre-defined and handcrafted distance thresholds for each category, and it is non-trivial to find optimal values

Don't you think sample_groups values are encoding apriori information?