Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Multiclass detection #8

Open
jrbtaylor opened this issue May 14, 2019 · 0 comments
Open

Multiclass detection #8

jrbtaylor opened this issue May 14, 2019 · 0 comments

Comments

@jrbtaylor
Copy link

Thanks for making the code public.

I noticed none of the example config files is set for detecting multiple classes (i.e. car, pedestrian and cyclist as 3 separate classes). I tried modifying one of the included configs by appending to "classes" and "num_clusters" in dataset_config but it crashed building the dataset. I think the related lines are 178-179 of avod.datasets.kitti.kitti_dataset:

            elif self.classes == ['Car', 'Pedestrian', 'Cyclist']:
                self.classes_name = 'All'

then 'All' is passed to get_anchor_info in avod.core.mini_batch_utils, which tries to find the preprocessed dataset files for 'All', which is not created in the preprocessing - instead it creates 3 directories for 'car', 'pedestrian' and 'cyclist'. I think there's either a mismatch between the preprocessing and training code, or I'm missing something in the config file that's unclear.

Is it possible to include one of the config files from your thesis work where you did multiclass detection on KITTI?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant