search

Gradiant / pyodi

Python Object Detection Insights
https://gradiant.github.io/pyodi/
Mozilla Public License 2.0
192 stars 16 forks source link

About clustering #164

Open cyh-123 opened 3 years ago

cyh-123 commented 3 years ago

Excuse me, I want to know how clustering is done. I don't see the call of core/clustering.py in the place where the plot is drawn? Did I understand it wrong?

mmeendez8 commented 3 years ago

Clustering is called in train_config_generation: https://github.com/Gradiant/pyodi/blob/master/pyodi/apps/train_config/train_config_generation.py#L190

We create an AnchorGenerator object from those clusters and that is what is plotted: https://github.com/Gradiant/pyodi/blob/master/pyodi/apps/train_config/train_config_generation.py#L206