Class Imbalance using SSD Network

I currently have a significant class imbalance between 2 classes and using the SSD network. I would like to use something similar to the class_weight parameter that detectnet has. Any information is appreciated.

SSD does not have similar parameter. For class imbalance, a better pretrained model plays an important role.

1 Like