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.

204 lines
8.9 KiB

  1. # Generated by the protocol buffer compiler. DO NOT EDIT!
  2. # source: object_detection/protos/post_processing.proto
  3. import sys
  4. _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
  5. from google.protobuf import descriptor as _descriptor
  6. from google.protobuf import message as _message
  7. from google.protobuf import reflection as _reflection
  8. from google.protobuf import symbol_database as _symbol_database
  9. from google.protobuf import descriptor_pb2
  10. # @@protoc_insertion_point(imports)
  11. _sym_db = _symbol_database.Default()
  12. from object_detection.protos import calibration_pb2 as object__detection_dot_protos_dot_calibration__pb2
  13. DESCRIPTOR = _descriptor.FileDescriptor(
  14. name='object_detection/protos/post_processing.proto',
  15. package='object_detection.protos',
  16. syntax='proto2',
  17. serialized_pb=_b('\n-object_detection/protos/post_processing.proto\x12\x17object_detection.protos\x1a)object_detection/protos/calibration.proto\"\x89\x02\n\x16\x42\x61tchNonMaxSuppression\x12\x1a\n\x0fscore_threshold\x18\x01 \x01(\x02:\x01\x30\x12\x1a\n\riou_threshold\x18\x02 \x01(\x02:\x03\x30.6\x12%\n\x18max_detections_per_class\x18\x03 \x01(\x05:\x03\x31\x30\x30\x12!\n\x14max_total_detections\x18\x05 \x01(\x05:\x03\x31\x30\x30\x12 \n\x11use_static_shapes\x18\x06 \x01(\x08:\x05\x66\x61lse\x12%\n\x16use_class_agnostic_nms\x18\x07 \x01(\x08:\x05\x66\x61lse\x12$\n\x19max_classes_per_detection\x18\x08 \x01(\x05:\x01\x31\"\xd9\x02\n\x0ePostProcessing\x12R\n\x19\x62\x61tch_non_max_suppression\x18\x01 \x01(\x0b\x32/.object_detection.protos.BatchNonMaxSuppression\x12Y\n\x0fscore_converter\x18\x02 \x01(\x0e\x32\x36.object_detection.protos.PostProcessing.ScoreConverter:\x08IDENTITY\x12\x16\n\x0blogit_scale\x18\x03 \x01(\x02:\x01\x31\x12\x46\n\x12\x63\x61libration_config\x18\x04 \x01(\x0b\x32*.object_detection.protos.CalibrationConfig\"8\n\x0eScoreConverter\x12\x0c\n\x08IDENTITY\x10\x00\x12\x0b\n\x07SIGMOID\x10\x01\x12\x0b\n\x07SOFTMAX\x10\x02')
  18. ,
  19. dependencies=[object__detection_dot_protos_dot_calibration__pb2.DESCRIPTOR,])
  20. _sym_db.RegisterFileDescriptor(DESCRIPTOR)
  21. _POSTPROCESSING_SCORECONVERTER = _descriptor.EnumDescriptor(
  22. name='ScoreConverter',
  23. full_name='object_detection.protos.PostProcessing.ScoreConverter',
  24. filename=None,
  25. file=DESCRIPTOR,
  26. values=[
  27. _descriptor.EnumValueDescriptor(
  28. name='IDENTITY', index=0, number=0,
  29. options=None,
  30. type=None),
  31. _descriptor.EnumValueDescriptor(
  32. name='SIGMOID', index=1, number=1,
  33. options=None,
  34. type=None),
  35. _descriptor.EnumValueDescriptor(
  36. name='SOFTMAX', index=2, number=2,
  37. options=None,
  38. type=None),
  39. ],
  40. containing_type=None,
  41. options=None,
  42. serialized_start=675,
  43. serialized_end=731,
  44. )
  45. _sym_db.RegisterEnumDescriptor(_POSTPROCESSING_SCORECONVERTER)
  46. _BATCHNONMAXSUPPRESSION = _descriptor.Descriptor(
  47. name='BatchNonMaxSuppression',
  48. full_name='object_detection.protos.BatchNonMaxSuppression',
  49. filename=None,
  50. file=DESCRIPTOR,
  51. containing_type=None,
  52. fields=[
  53. _descriptor.FieldDescriptor(
  54. name='score_threshold', full_name='object_detection.protos.BatchNonMaxSuppression.score_threshold', index=0,
  55. number=1, type=2, cpp_type=6, label=1,
  56. has_default_value=True, default_value=float(0),
  57. message_type=None, enum_type=None, containing_type=None,
  58. is_extension=False, extension_scope=None,
  59. options=None),
  60. _descriptor.FieldDescriptor(
  61. name='iou_threshold', full_name='object_detection.protos.BatchNonMaxSuppression.iou_threshold', index=1,
  62. number=2, type=2, cpp_type=6, label=1,
  63. has_default_value=True, default_value=float(0.6),
  64. message_type=None, enum_type=None, containing_type=None,
  65. is_extension=False, extension_scope=None,
  66. options=None),
  67. _descriptor.FieldDescriptor(
  68. name='max_detections_per_class', full_name='object_detection.protos.BatchNonMaxSuppression.max_detections_per_class', index=2,
  69. number=3, type=5, cpp_type=1, label=1,
  70. has_default_value=True, default_value=100,
  71. message_type=None, enum_type=None, containing_type=None,
  72. is_extension=False, extension_scope=None,
  73. options=None),
  74. _descriptor.FieldDescriptor(
  75. name='max_total_detections', full_name='object_detection.protos.BatchNonMaxSuppression.max_total_detections', index=3,
  76. number=5, type=5, cpp_type=1, label=1,
  77. has_default_value=True, default_value=100,
  78. message_type=None, enum_type=None, containing_type=None,
  79. is_extension=False, extension_scope=None,
  80. options=None),
  81. _descriptor.FieldDescriptor(
  82. name='use_static_shapes', full_name='object_detection.protos.BatchNonMaxSuppression.use_static_shapes', index=4,
  83. number=6, type=8, cpp_type=7, label=1,
  84. has_default_value=True, default_value=False,
  85. message_type=None, enum_type=None, containing_type=None,
  86. is_extension=False, extension_scope=None,
  87. options=None),
  88. _descriptor.FieldDescriptor(
  89. name='use_class_agnostic_nms', full_name='object_detection.protos.BatchNonMaxSuppression.use_class_agnostic_nms', index=5,
  90. number=7, type=8, cpp_type=7, label=1,
  91. has_default_value=True, default_value=False,
  92. message_type=None, enum_type=None, containing_type=None,
  93. is_extension=False, extension_scope=None,
  94. options=None),
  95. _descriptor.FieldDescriptor(
  96. name='max_classes_per_detection', full_name='object_detection.protos.BatchNonMaxSuppression.max_classes_per_detection', index=6,
  97. number=8, type=5, cpp_type=1, label=1,
  98. has_default_value=True, default_value=1,
  99. message_type=None, enum_type=None, containing_type=None,
  100. is_extension=False, extension_scope=None,
  101. options=None),
  102. ],
  103. extensions=[
  104. ],
  105. nested_types=[],
  106. enum_types=[
  107. ],
  108. options=None,
  109. is_extendable=False,
  110. syntax='proto2',
  111. extension_ranges=[],
  112. oneofs=[
  113. ],
  114. serialized_start=118,
  115. serialized_end=383,
  116. )
  117. _POSTPROCESSING = _descriptor.Descriptor(
  118. name='PostProcessing',
  119. full_name='object_detection.protos.PostProcessing',
  120. filename=None,
  121. file=DESCRIPTOR,
  122. containing_type=None,
  123. fields=[
  124. _descriptor.FieldDescriptor(
  125. name='batch_non_max_suppression', full_name='object_detection.protos.PostProcessing.batch_non_max_suppression', index=0,
  126. number=1, type=11, cpp_type=10, label=1,
  127. has_default_value=False, default_value=None,
  128. message_type=None, enum_type=None, containing_type=None,
  129. is_extension=False, extension_scope=None,
  130. options=None),
  131. _descriptor.FieldDescriptor(
  132. name='score_converter', full_name='object_detection.protos.PostProcessing.score_converter', index=1,
  133. number=2, type=14, cpp_type=8, label=1,
  134. has_default_value=True, default_value=0,
  135. message_type=None, enum_type=None, containing_type=None,
  136. is_extension=False, extension_scope=None,
  137. options=None),
  138. _descriptor.FieldDescriptor(
  139. name='logit_scale', full_name='object_detection.protos.PostProcessing.logit_scale', index=2,
  140. number=3, type=2, cpp_type=6, label=1,
  141. has_default_value=True, default_value=float(1),
  142. message_type=None, enum_type=None, containing_type=None,
  143. is_extension=False, extension_scope=None,
  144. options=None),
  145. _descriptor.FieldDescriptor(
  146. name='calibration_config', full_name='object_detection.protos.PostProcessing.calibration_config', index=3,
  147. number=4, type=11, cpp_type=10, label=1,
  148. has_default_value=False, default_value=None,
  149. message_type=None, enum_type=None, containing_type=None,
  150. is_extension=False, extension_scope=None,
  151. options=None),
  152. ],
  153. extensions=[
  154. ],
  155. nested_types=[],
  156. enum_types=[
  157. _POSTPROCESSING_SCORECONVERTER,
  158. ],
  159. options=None,
  160. is_extendable=False,
  161. syntax='proto2',
  162. extension_ranges=[],
  163. oneofs=[
  164. ],
  165. serialized_start=386,
  166. serialized_end=731,
  167. )
  168. _POSTPROCESSING.fields_by_name['batch_non_max_suppression'].message_type = _BATCHNONMAXSUPPRESSION
  169. _POSTPROCESSING.fields_by_name['score_converter'].enum_type = _POSTPROCESSING_SCORECONVERTER
  170. _POSTPROCESSING.fields_by_name['calibration_config'].message_type = object__detection_dot_protos_dot_calibration__pb2._CALIBRATIONCONFIG
  171. _POSTPROCESSING_SCORECONVERTER.containing_type = _POSTPROCESSING
  172. DESCRIPTOR.message_types_by_name['BatchNonMaxSuppression'] = _BATCHNONMAXSUPPRESSION
  173. DESCRIPTOR.message_types_by_name['PostProcessing'] = _POSTPROCESSING
  174. BatchNonMaxSuppression = _reflection.GeneratedProtocolMessageType('BatchNonMaxSuppression', (_message.Message,), dict(
  175. DESCRIPTOR = _BATCHNONMAXSUPPRESSION,
  176. __module__ = 'object_detection.protos.post_processing_pb2'
  177. # @@protoc_insertion_point(class_scope:object_detection.protos.BatchNonMaxSuppression)
  178. ))
  179. _sym_db.RegisterMessage(BatchNonMaxSuppression)
  180. PostProcessing = _reflection.GeneratedProtocolMessageType('PostProcessing', (_message.Message,), dict(
  181. DESCRIPTOR = _POSTPROCESSING,
  182. __module__ = 'object_detection.protos.post_processing_pb2'
  183. # @@protoc_insertion_point(class_scope:object_detection.protos.PostProcessing)
  184. ))
  185. _sym_db.RegisterMessage(PostProcessing)
  186. # @@protoc_insertion_point(module_scope)