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.

266 lines
15 KiB

  1. # Generated by the protocol buffer compiler. DO NOT EDIT!
  2. # source: object_detection/protos/train.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 optimizer_pb2 as object__detection_dot_protos_dot_optimizer__pb2
  13. from object_detection.protos import preprocessor_pb2 as object__detection_dot_protos_dot_preprocessor__pb2
  14. DESCRIPTOR = _descriptor.FileDescriptor(
  15. name='object_detection/protos/train.proto',
  16. package='object_detection.protos',
  17. syntax='proto2',
  18. serialized_pb=_b('\n#object_detection/protos/train.proto\x12\x17object_detection.protos\x1a\'object_detection/protos/optimizer.proto\x1a*object_detection/protos/preprocessor.proto\"\xa1\x08\n\x0bTrainConfig\x12\x16\n\nbatch_size\x18\x01 \x01(\r:\x02\x33\x32\x12M\n\x19\x64\x61ta_augmentation_options\x18\x02 \x03(\x0b\x32*.object_detection.protos.PreprocessingStep\x12\x1c\n\rsync_replicas\x18\x03 \x01(\x08:\x05\x66\x61lse\x12,\n\x1dkeep_checkpoint_every_n_hours\x18\x04 \x01(\x02:\x05\x31\x30\x30\x30\x30\x12\x35\n\toptimizer\x18\x05 \x01(\x0b\x32\".object_detection.protos.Optimizer\x12$\n\x19gradient_clipping_by_norm\x18\x06 \x01(\x02:\x01\x30\x12\x1e\n\x14\x66ine_tune_checkpoint\x18\x07 \x01(\t:\x00\x12#\n\x19\x66ine_tune_checkpoint_type\x18\x16 \x01(\t:\x00\x12,\n\x19\x66rom_detection_checkpoint\x18\x08 \x01(\x08:\x05\x66\x61lseB\x02\x18\x01\x12\x31\n\"load_all_detection_checkpoint_vars\x18\x13 \x01(\x08:\x05\x66\x61lse\x12\x14\n\tnum_steps\x18\t \x01(\r:\x01\x30\x12\x1f\n\x13startup_delay_steps\x18\n \x01(\x02:\x02\x31\x35\x12\x1f\n\x14\x62ias_grad_multiplier\x18\x0b \x01(\x02:\x01\x30\x12\"\n\x1aupdate_trainable_variables\x18\x19 \x03(\t\x12\x18\n\x10\x66reeze_variables\x18\x0c \x03(\t\x12 \n\x15replicas_to_aggregate\x18\r \x01(\x05:\x01\x31\x12%\n\x14\x62\x61tch_queue_capacity\x18\x0e \x01(\x05:\x03\x31\x35\x30\x42\x02\x18\x01\x12&\n\x17num_batch_queue_threads\x18\x0f \x01(\x05:\x01\x38\x42\x02\x18\x01\x12&\n\x17prefetch_queue_capacity\x18\x10 \x01(\x05:\x01\x35\x42\x02\x18\x01\x12)\n\x1amerge_multiple_label_boxes\x18\x11 \x01(\x08:\x05\x66\x61lse\x12$\n\x15use_multiclass_scores\x18\x18 \x01(\x08:\x05\x66\x61lse\x12%\n\x17\x61\x64\x64_regularization_loss\x18\x12 \x01(\x08:\x04true\x12$\n\x13max_number_of_boxes\x18\x14 \x01(\x05:\x03\x31\x30\x30\x42\x02\x18\x01\x12\'\n\x19unpad_groundtruth_tensors\x18\x15 \x01(\x08:\x04true\x12%\n\x16retain_original_images\x18\x17 \x01(\x08:\x05\x66\x61lse\x12\x1b\n\x0cuse_bfloat16\x18\x1a \x01(\x08:\x05\x66\x61lse\x12\"\n\x13summarize_gradients\x18\x1b \x01(\x08:\x05\x66\x61lse')
  19. ,
  20. dependencies=[object__detection_dot_protos_dot_optimizer__pb2.DESCRIPTOR,object__detection_dot_protos_dot_preprocessor__pb2.DESCRIPTOR,])
  21. _sym_db.RegisterFileDescriptor(DESCRIPTOR)
  22. _TRAINCONFIG = _descriptor.Descriptor(
  23. name='TrainConfig',
  24. full_name='object_detection.protos.TrainConfig',
  25. filename=None,
  26. file=DESCRIPTOR,
  27. containing_type=None,
  28. fields=[
  29. _descriptor.FieldDescriptor(
  30. name='batch_size', full_name='object_detection.protos.TrainConfig.batch_size', index=0,
  31. number=1, type=13, cpp_type=3, label=1,
  32. has_default_value=True, default_value=32,
  33. message_type=None, enum_type=None, containing_type=None,
  34. is_extension=False, extension_scope=None,
  35. options=None),
  36. _descriptor.FieldDescriptor(
  37. name='data_augmentation_options', full_name='object_detection.protos.TrainConfig.data_augmentation_options', index=1,
  38. number=2, type=11, cpp_type=10, label=3,
  39. has_default_value=False, default_value=[],
  40. message_type=None, enum_type=None, containing_type=None,
  41. is_extension=False, extension_scope=None,
  42. options=None),
  43. _descriptor.FieldDescriptor(
  44. name='sync_replicas', full_name='object_detection.protos.TrainConfig.sync_replicas', index=2,
  45. number=3, type=8, cpp_type=7, label=1,
  46. has_default_value=True, default_value=False,
  47. message_type=None, enum_type=None, containing_type=None,
  48. is_extension=False, extension_scope=None,
  49. options=None),
  50. _descriptor.FieldDescriptor(
  51. name='keep_checkpoint_every_n_hours', full_name='object_detection.protos.TrainConfig.keep_checkpoint_every_n_hours', index=3,
  52. number=4, type=2, cpp_type=6, label=1,
  53. has_default_value=True, default_value=float(10000),
  54. message_type=None, enum_type=None, containing_type=None,
  55. is_extension=False, extension_scope=None,
  56. options=None),
  57. _descriptor.FieldDescriptor(
  58. name='optimizer', full_name='object_detection.protos.TrainConfig.optimizer', index=4,
  59. number=5, type=11, cpp_type=10, label=1,
  60. has_default_value=False, default_value=None,
  61. message_type=None, enum_type=None, containing_type=None,
  62. is_extension=False, extension_scope=None,
  63. options=None),
  64. _descriptor.FieldDescriptor(
  65. name='gradient_clipping_by_norm', full_name='object_detection.protos.TrainConfig.gradient_clipping_by_norm', index=5,
  66. number=6, type=2, cpp_type=6, label=1,
  67. has_default_value=True, default_value=float(0),
  68. message_type=None, enum_type=None, containing_type=None,
  69. is_extension=False, extension_scope=None,
  70. options=None),
  71. _descriptor.FieldDescriptor(
  72. name='fine_tune_checkpoint', full_name='object_detection.protos.TrainConfig.fine_tune_checkpoint', index=6,
  73. number=7, type=9, cpp_type=9, label=1,
  74. has_default_value=True, default_value=_b("").decode('utf-8'),
  75. message_type=None, enum_type=None, containing_type=None,
  76. is_extension=False, extension_scope=None,
  77. options=None),
  78. _descriptor.FieldDescriptor(
  79. name='fine_tune_checkpoint_type', full_name='object_detection.protos.TrainConfig.fine_tune_checkpoint_type', index=7,
  80. number=22, type=9, cpp_type=9, label=1,
  81. has_default_value=True, default_value=_b("").decode('utf-8'),
  82. message_type=None, enum_type=None, containing_type=None,
  83. is_extension=False, extension_scope=None,
  84. options=None),
  85. _descriptor.FieldDescriptor(
  86. name='from_detection_checkpoint', full_name='object_detection.protos.TrainConfig.from_detection_checkpoint', index=8,
  87. number=8, type=8, cpp_type=7, label=1,
  88. has_default_value=True, default_value=False,
  89. message_type=None, enum_type=None, containing_type=None,
  90. is_extension=False, extension_scope=None,
  91. options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))),
  92. _descriptor.FieldDescriptor(
  93. name='load_all_detection_checkpoint_vars', full_name='object_detection.protos.TrainConfig.load_all_detection_checkpoint_vars', index=9,
  94. number=19, type=8, cpp_type=7, label=1,
  95. has_default_value=True, default_value=False,
  96. message_type=None, enum_type=None, containing_type=None,
  97. is_extension=False, extension_scope=None,
  98. options=None),
  99. _descriptor.FieldDescriptor(
  100. name='num_steps', full_name='object_detection.protos.TrainConfig.num_steps', index=10,
  101. number=9, type=13, cpp_type=3, label=1,
  102. has_default_value=True, default_value=0,
  103. message_type=None, enum_type=None, containing_type=None,
  104. is_extension=False, extension_scope=None,
  105. options=None),
  106. _descriptor.FieldDescriptor(
  107. name='startup_delay_steps', full_name='object_detection.protos.TrainConfig.startup_delay_steps', index=11,
  108. number=10, type=2, cpp_type=6, label=1,
  109. has_default_value=True, default_value=float(15),
  110. message_type=None, enum_type=None, containing_type=None,
  111. is_extension=False, extension_scope=None,
  112. options=None),
  113. _descriptor.FieldDescriptor(
  114. name='bias_grad_multiplier', full_name='object_detection.protos.TrainConfig.bias_grad_multiplier', index=12,
  115. number=11, type=2, cpp_type=6, label=1,
  116. has_default_value=True, default_value=float(0),
  117. message_type=None, enum_type=None, containing_type=None,
  118. is_extension=False, extension_scope=None,
  119. options=None),
  120. _descriptor.FieldDescriptor(
  121. name='update_trainable_variables', full_name='object_detection.protos.TrainConfig.update_trainable_variables', index=13,
  122. number=25, type=9, cpp_type=9, label=3,
  123. has_default_value=False, default_value=[],
  124. message_type=None, enum_type=None, containing_type=None,
  125. is_extension=False, extension_scope=None,
  126. options=None),
  127. _descriptor.FieldDescriptor(
  128. name='freeze_variables', full_name='object_detection.protos.TrainConfig.freeze_variables', index=14,
  129. number=12, type=9, cpp_type=9, label=3,
  130. has_default_value=False, default_value=[],
  131. message_type=None, enum_type=None, containing_type=None,
  132. is_extension=False, extension_scope=None,
  133. options=None),
  134. _descriptor.FieldDescriptor(
  135. name='replicas_to_aggregate', full_name='object_detection.protos.TrainConfig.replicas_to_aggregate', index=15,
  136. number=13, type=5, cpp_type=1, label=1,
  137. has_default_value=True, default_value=1,
  138. message_type=None, enum_type=None, containing_type=None,
  139. is_extension=False, extension_scope=None,
  140. options=None),
  141. _descriptor.FieldDescriptor(
  142. name='batch_queue_capacity', full_name='object_detection.protos.TrainConfig.batch_queue_capacity', index=16,
  143. number=14, type=5, cpp_type=1, label=1,
  144. has_default_value=True, default_value=150,
  145. message_type=None, enum_type=None, containing_type=None,
  146. is_extension=False, extension_scope=None,
  147. options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))),
  148. _descriptor.FieldDescriptor(
  149. name='num_batch_queue_threads', full_name='object_detection.protos.TrainConfig.num_batch_queue_threads', index=17,
  150. number=15, type=5, cpp_type=1, label=1,
  151. has_default_value=True, default_value=8,
  152. message_type=None, enum_type=None, containing_type=None,
  153. is_extension=False, extension_scope=None,
  154. options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))),
  155. _descriptor.FieldDescriptor(
  156. name='prefetch_queue_capacity', full_name='object_detection.protos.TrainConfig.prefetch_queue_capacity', index=18,
  157. number=16, type=5, cpp_type=1, label=1,
  158. has_default_value=True, default_value=5,
  159. message_type=None, enum_type=None, containing_type=None,
  160. is_extension=False, extension_scope=None,
  161. options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))),
  162. _descriptor.FieldDescriptor(
  163. name='merge_multiple_label_boxes', full_name='object_detection.protos.TrainConfig.merge_multiple_label_boxes', index=19,
  164. number=17, type=8, cpp_type=7, label=1,
  165. has_default_value=True, default_value=False,
  166. message_type=None, enum_type=None, containing_type=None,
  167. is_extension=False, extension_scope=None,
  168. options=None),
  169. _descriptor.FieldDescriptor(
  170. name='use_multiclass_scores', full_name='object_detection.protos.TrainConfig.use_multiclass_scores', index=20,
  171. number=24, type=8, cpp_type=7, label=1,
  172. has_default_value=True, default_value=False,
  173. message_type=None, enum_type=None, containing_type=None,
  174. is_extension=False, extension_scope=None,
  175. options=None),
  176. _descriptor.FieldDescriptor(
  177. name='add_regularization_loss', full_name='object_detection.protos.TrainConfig.add_regularization_loss', index=21,
  178. number=18, type=8, cpp_type=7, label=1,
  179. has_default_value=True, default_value=True,
  180. message_type=None, enum_type=None, containing_type=None,
  181. is_extension=False, extension_scope=None,
  182. options=None),
  183. _descriptor.FieldDescriptor(
  184. name='max_number_of_boxes', full_name='object_detection.protos.TrainConfig.max_number_of_boxes', index=22,
  185. number=20, type=5, cpp_type=1, label=1,
  186. has_default_value=True, default_value=100,
  187. message_type=None, enum_type=None, containing_type=None,
  188. is_extension=False, extension_scope=None,
  189. options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))),
  190. _descriptor.FieldDescriptor(
  191. name='unpad_groundtruth_tensors', full_name='object_detection.protos.TrainConfig.unpad_groundtruth_tensors', index=23,
  192. number=21, type=8, cpp_type=7, label=1,
  193. has_default_value=True, default_value=True,
  194. message_type=None, enum_type=None, containing_type=None,
  195. is_extension=False, extension_scope=None,
  196. options=None),
  197. _descriptor.FieldDescriptor(
  198. name='retain_original_images', full_name='object_detection.protos.TrainConfig.retain_original_images', index=24,
  199. number=23, type=8, cpp_type=7, label=1,
  200. has_default_value=True, default_value=False,
  201. message_type=None, enum_type=None, containing_type=None,
  202. is_extension=False, extension_scope=None,
  203. options=None),
  204. _descriptor.FieldDescriptor(
  205. name='use_bfloat16', full_name='object_detection.protos.TrainConfig.use_bfloat16', index=25,
  206. number=26, type=8, cpp_type=7, label=1,
  207. has_default_value=True, default_value=False,
  208. message_type=None, enum_type=None, containing_type=None,
  209. is_extension=False, extension_scope=None,
  210. options=None),
  211. _descriptor.FieldDescriptor(
  212. name='summarize_gradients', full_name='object_detection.protos.TrainConfig.summarize_gradients', index=26,
  213. number=27, type=8, cpp_type=7, label=1,
  214. has_default_value=True, default_value=False,
  215. message_type=None, enum_type=None, containing_type=None,
  216. is_extension=False, extension_scope=None,
  217. options=None),
  218. ],
  219. extensions=[
  220. ],
  221. nested_types=[],
  222. enum_types=[
  223. ],
  224. options=None,
  225. is_extendable=False,
  226. syntax='proto2',
  227. extension_ranges=[],
  228. oneofs=[
  229. ],
  230. serialized_start=150,
  231. serialized_end=1207,
  232. )
  233. _TRAINCONFIG.fields_by_name['data_augmentation_options'].message_type = object__detection_dot_protos_dot_preprocessor__pb2._PREPROCESSINGSTEP
  234. _TRAINCONFIG.fields_by_name['optimizer'].message_type = object__detection_dot_protos_dot_optimizer__pb2._OPTIMIZER
  235. DESCRIPTOR.message_types_by_name['TrainConfig'] = _TRAINCONFIG
  236. TrainConfig = _reflection.GeneratedProtocolMessageType('TrainConfig', (_message.Message,), dict(
  237. DESCRIPTOR = _TRAINCONFIG,
  238. __module__ = 'object_detection.protos.train_pb2'
  239. # @@protoc_insertion_point(class_scope:object_detection.protos.TrainConfig)
  240. ))
  241. _sym_db.RegisterMessage(TrainConfig)
  242. _TRAINCONFIG.fields_by_name['from_detection_checkpoint'].has_options = True
  243. _TRAINCONFIG.fields_by_name['from_detection_checkpoint']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))
  244. _TRAINCONFIG.fields_by_name['batch_queue_capacity'].has_options = True
  245. _TRAINCONFIG.fields_by_name['batch_queue_capacity']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))
  246. _TRAINCONFIG.fields_by_name['num_batch_queue_threads'].has_options = True
  247. _TRAINCONFIG.fields_by_name['num_batch_queue_threads']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))
  248. _TRAINCONFIG.fields_by_name['prefetch_queue_capacity'].has_options = True
  249. _TRAINCONFIG.fields_by_name['prefetch_queue_capacity']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))
  250. _TRAINCONFIG.fields_by_name['max_number_of_boxes'].has_options = True
  251. _TRAINCONFIG.fields_by_name['max_number_of_boxes']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001'))
  252. # @@protoc_insertion_point(module_scope)