You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

29 lines
1.1 KiB

6 years ago
  1. syntax = "proto2";
  2. package object_detection.protos;
  3. // Configuration proto for ArgMaxMatcher. See
  4. // matchers/argmax_matcher.py for details.
  5. message ArgMaxMatcher {
  6. // Threshold for positive matches.
  7. optional float matched_threshold = 1 [default = 0.5];
  8. // Threshold for negative matches.
  9. optional float unmatched_threshold = 2 [default = 0.5];
  10. // Whether to construct ArgMaxMatcher without thresholds.
  11. optional bool ignore_thresholds = 3 [default = false];
  12. // If True then negative matches are the ones below the unmatched_threshold,
  13. // whereas ignored matches are in between the matched and umatched
  14. // threshold. If False, then negative matches are in between the matched
  15. // and unmatched threshold, and everything lower than unmatched is ignored.
  16. optional bool negatives_lower_than_unmatched = 4 [default = true];
  17. // Whether to ensure each row is matched to at least one column.
  18. optional bool force_match_for_each_row = 5 [default = false];
  19. // Force constructed match objects to use matrix multiplication based gather
  20. // instead of standard tf.gather
  21. optional bool use_matmul_gather = 6 [default = false];
  22. }