Adjust NonMaxSuppression score_threshold in build_detection_graph for ssd networks #19
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
It took me quite some time to figure this out, so I think people would be happy to have this automatically done for them when running the
build_detection_graph
function. The threshold can be adjusted by passing in an argument. With this fix, it should be possible to get the described performance without any tweaks to the models.I used the warnings module to inform the user about if the value was changed or if it was not possible to change it. I can change that if necessary.