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

Additional API control over useful SSD training / deploy parameters #651

Merged
merged 3 commits into from Oct 17, 2019

Conversation

beniz
Copy link
Collaborator

@beniz beniz commented Oct 15, 2019

This PR adds controls over the following SSD parameters at service/model creation time (from template), the mllib object:

  • ssd_expand_prob: probability of expanding the image (to improve detection of small/very small objects)
  • ssd_max_expand_ratio: bbox zoom out ratio
  • ssd_mining_type: "HARD_EXAMPLE" or "MAX_NEGATIVE"
  • ssd_neg_pos_ratio: ratio of negative sampled examples wrt positive examples (bbox)
  • ssd_neg_overlap: max overlap of negative samples with positive samples (bbox)
  • ssd_keep_top_k: keep k examples after nms has finished

The default values remain that of every SSD template, and are only overridden when specified through API.

Copy link
Contributor

@fantes fantes left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ok for me

@beniz beniz merged commit 2da03ac into master Oct 17, 2019
@beniz beniz deleted the ssd_options branch April 22, 2020 09:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants