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

Why isn't negative sampling used during the training of ssd detector? #333

Open
allgrom opened this issue Sep 24, 2018 · 2 comments
Open
Labels
question Further information is requested

Comments

@allgrom
Copy link

allgrom commented Sep 24, 2018

Hi!
Why isn't negative sampling used during the training of ssd detector?

self._target_generator = SSDTargetGenerator(
            iou_thresh=iou_thresh, stds=box_norm, negative_mining_ratio=-1, **kwargs)
@zhreshold
Copy link
Member

TLDR. It's moved to MultiBoxLoss.
Reason: OHEM is dependent on prediction value, but we want to pre-compute all targets using cpu cores during data loading. Instead we can use cpu to compute all targets and mask out negative samples in MultiBoxLoss

@zhreshold zhreshold added the question Further information is requested label Sep 24, 2018
@allgrom
Copy link
Author

allgrom commented Sep 25, 2018

Thank you!

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

No branches or pull requests

2 participants