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.

190 lines
8.3 KiB

6 years ago
  1. # Generated by the protocol buffer compiler. DO NOT EDIT!
  2. # source: object_detection/protos/post_processing.proto
  3. import sys,getpass
  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. # @@protoc_insertion_point(imports)
  10. _sym_db = _symbol_database.Default()
  11. from object_detection.protos import calibration_pb2 as object__detection_dot_protos_dot_calibration__pb2
  12. DESCRIPTOR = _descriptor.FileDescriptor(
  13. name='object_detection/protos/post_processing.proto',
  14. package='object_detection.protos',
  15. syntax='proto2',
  16. serialized_options=None,
  17. serialized_pb=_b('\n-object_detection/protos/post_processing.proto\x12\x17object_detection.protos\x1a)object_detection/protos/calibration.proto\"\xbc\x01\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\"\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. _POSTPROCESSING_SCORECONVERTER = _descriptor.EnumDescriptor(
  21. name='ScoreConverter',
  22. full_name='object_detection.protos.PostProcessing.ScoreConverter',
  23. filename=None,
  24. file=DESCRIPTOR,
  25. values=[
  26. _descriptor.EnumValueDescriptor(
  27. name='IDENTITY', index=0, number=0,
  28. serialized_options=None,
  29. type=None),
  30. _descriptor.EnumValueDescriptor(
  31. name='SIGMOID', index=1, number=1,
  32. serialized_options=None,
  33. type=None),
  34. _descriptor.EnumValueDescriptor(
  35. name='SOFTMAX', index=2, number=2,
  36. serialized_options=None,
  37. type=None),
  38. ],
  39. containing_type=None,
  40. serialized_options=None,
  41. serialized_start=598,
  42. serialized_end=654,
  43. )
  44. _sym_db.RegisterEnumDescriptor(_POSTPROCESSING_SCORECONVERTER)
  45. _BATCHNONMAXSUPPRESSION = _descriptor.Descriptor(
  46. name='BatchNonMaxSuppression',
  47. full_name='object_detection.protos.BatchNonMaxSuppression',
  48. filename=None,
  49. file=DESCRIPTOR,
  50. containing_type=None,
  51. fields=[
  52. _descriptor.FieldDescriptor(
  53. name='score_threshold', full_name='object_detection.protos.BatchNonMaxSuppression.score_threshold', index=0,
  54. number=1, type=2, cpp_type=6, label=1,
  55. has_default_value=True, default_value=float(0),
  56. message_type=None, enum_type=None, containing_type=None,
  57. is_extension=False, extension_scope=None,
  58. serialized_options=None, file=DESCRIPTOR),
  59. _descriptor.FieldDescriptor(
  60. name='iou_threshold', full_name='object_detection.protos.BatchNonMaxSuppression.iou_threshold', index=1,
  61. number=2, type=2, cpp_type=6, label=1,
  62. has_default_value=True, default_value=float(0.6),
  63. message_type=None, enum_type=None, containing_type=None,
  64. is_extension=False, extension_scope=None,
  65. serialized_options=None, file=DESCRIPTOR),
  66. _descriptor.FieldDescriptor(
  67. name='max_detections_per_class', full_name='object_detection.protos.BatchNonMaxSuppression.max_detections_per_class', index=2,
  68. number=3, type=5, cpp_type=1, label=1,
  69. has_default_value=True, default_value=100,
  70. message_type=None, enum_type=None, containing_type=None,
  71. is_extension=False, extension_scope=None,
  72. serialized_options=None, file=DESCRIPTOR),
  73. _descriptor.FieldDescriptor(
  74. name='max_total_detections', full_name='object_detection.protos.BatchNonMaxSuppression.max_total_detections', index=3,
  75. number=5, type=5, cpp_type=1, label=1,
  76. has_default_value=True, default_value=100,
  77. message_type=None, enum_type=None, containing_type=None,
  78. is_extension=False, extension_scope=None,
  79. serialized_options=None, file=DESCRIPTOR),
  80. _descriptor.FieldDescriptor(
  81. name='use_static_shapes', full_name='object_detection.protos.BatchNonMaxSuppression.use_static_shapes', index=4,
  82. number=6, type=8, cpp_type=7, label=1,
  83. has_default_value=True, default_value=False,
  84. message_type=None, enum_type=None, containing_type=None,
  85. is_extension=False, extension_scope=None,
  86. serialized_options=None, file=DESCRIPTOR),
  87. ],
  88. extensions=[
  89. ],
  90. nested_types=[],
  91. enum_types=[
  92. ],
  93. serialized_options=None,
  94. is_extendable=False,
  95. syntax='proto2',
  96. extension_ranges=[],
  97. oneofs=[
  98. ],
  99. serialized_start=118,
  100. serialized_end=306,
  101. )
  102. _POSTPROCESSING = _descriptor.Descriptor(
  103. name='PostProcessing',
  104. full_name='object_detection.protos.PostProcessing',
  105. filename=None,
  106. file=DESCRIPTOR,
  107. containing_type=None,
  108. fields=[
  109. _descriptor.FieldDescriptor(
  110. name='batch_non_max_suppression', full_name='object_detection.protos.PostProcessing.batch_non_max_suppression', index=0,
  111. number=1, type=11, cpp_type=10, label=1,
  112. has_default_value=False, default_value=None,
  113. message_type=None, enum_type=None, containing_type=None,
  114. is_extension=False, extension_scope=None,
  115. serialized_options=None, file=DESCRIPTOR),
  116. _descriptor.FieldDescriptor(
  117. name='score_converter', full_name='object_detection.protos.PostProcessing.score_converter', index=1,
  118. number=2, type=14, cpp_type=8, label=1,
  119. has_default_value=True, default_value=0,
  120. message_type=None, enum_type=None, containing_type=None,
  121. is_extension=False, extension_scope=None,
  122. serialized_options=None, file=DESCRIPTOR),
  123. _descriptor.FieldDescriptor(
  124. name='logit_scale', full_name='object_detection.protos.PostProcessing.logit_scale', index=2,
  125. number=3, type=2, cpp_type=6, label=1,
  126. has_default_value=True, default_value=float(1),
  127. message_type=None, enum_type=None, containing_type=None,
  128. is_extension=False, extension_scope=None,
  129. serialized_options=None, file=DESCRIPTOR),
  130. _descriptor.FieldDescriptor(
  131. name='calibration_config', full_name='object_detection.protos.PostProcessing.calibration_config', index=3,
  132. number=4, type=11, cpp_type=10, label=1,
  133. has_default_value=False, default_value=None,
  134. message_type=None, enum_type=None, containing_type=None,
  135. is_extension=False, extension_scope=None,
  136. serialized_options=None, file=DESCRIPTOR),
  137. ],
  138. extensions=[
  139. ],
  140. nested_types=[],
  141. enum_types=[
  142. _POSTPROCESSING_SCORECONVERTER,
  143. ],
  144. serialized_options=None,
  145. is_extendable=False,
  146. syntax='proto2',
  147. extension_ranges=[],
  148. oneofs=[
  149. ],
  150. serialized_start=309,
  151. serialized_end=654,
  152. )
  153. _POSTPROCESSING.fields_by_name['batch_non_max_suppression'].message_type = _BATCHNONMAXSUPPRESSION
  154. _POSTPROCESSING.fields_by_name['score_converter'].enum_type = _POSTPROCESSING_SCORECONVERTER
  155. _POSTPROCESSING.fields_by_name['calibration_config'].message_type = object__detection_dot_protos_dot_calibration__pb2._CALIBRATIONCONFIG
  156. _POSTPROCESSING_SCORECONVERTER.containing_type = _POSTPROCESSING
  157. DESCRIPTOR.message_types_by_name['BatchNonMaxSuppression'] = _BATCHNONMAXSUPPRESSION
  158. DESCRIPTOR.message_types_by_name['PostProcessing'] = _POSTPROCESSING
  159. _sym_db.RegisterFileDescriptor(DESCRIPTOR)
  160. BatchNonMaxSuppression = _reflection.GeneratedProtocolMessageType('BatchNonMaxSuppression', (_message.Message,), dict(
  161. DESCRIPTOR = _BATCHNONMAXSUPPRESSION,
  162. __module__ = 'object_detection.protos.post_processing_pb2'
  163. # @@protoc_insertion_point(class_scope:object_detection.protos.BatchNonMaxSuppression)
  164. ))
  165. _sym_db.RegisterMessage(BatchNonMaxSuppression)
  166. PostProcessing = _reflection.GeneratedProtocolMessageType('PostProcessing', (_message.Message,), dict(
  167. DESCRIPTOR = _POSTPROCESSING,
  168. __module__ = 'object_detection.protos.post_processing_pb2'
  169. # @@protoc_insertion_point(class_scope:object_detection.protos.PostProcessing)
  170. ))
  171. _sym_db.RegisterMessage(PostProcessing)
  172. # @@protoc_insertion_point(module_scope)