# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: nvidia_tao_tf1/cv/yolo_v3/proto/dataset_config.proto

import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()




DESCRIPTOR = _descriptor.FileDescriptor(
  name='nvidia_tao_tf1/cv/yolo_v3/proto/dataset_config.proto',
  package='',
  syntax='proto3',
  serialized_options=None,
  serialized_pb=_b('\n4nvidia_tao_tf1/cv/yolo_v3/proto/dataset_config.proto\"\xa5\x01\n\x10YOLOv3DataSource\x12\x1e\n\x14label_directory_path\x18\x01 \x01(\tH\x00\x12\x18\n\x0etfrecords_path\x18\x02 \x01(\tH\x00\x12\x1c\n\x14image_directory_path\x18\x03 \x01(\t\x12\x11\n\troot_path\x18\x04 \x01(\t\x12\x15\n\rsource_weight\x18\x05 \x01(\x02\x42\x0f\n\rlabels_format\"\xf7\x02\n\x13YOLOv3DatasetConfig\x12\'\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32\x11.YOLOv3DataSource\x12J\n\x14target_class_mapping\x18\x02 \x03(\x0b\x32,.YOLOv3DatasetConfig.TargetClassMappingEntry\x12\x17\n\x0fvalidation_fold\x18\x04 \x01(\r\x12\x32\n\x17validation_data_sources\x18\x03 \x03(\x0b\x32\x11.YOLOv3DataSource\x12%\n\x1dinclude_difficult_in_training\x18\x07 \x01(\x08\x12\x0c\n\x04type\x18\x05 \x01(\t\x12\x17\n\x0fimage_extension\x18\x06 \x01(\t\x12\x15\n\ris_monochrome\x18\x08 \x01(\x08\x1a\x39\n\x17TargetClassMappingEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x62\x06proto3')
)




_YOLOV3DATASOURCE = _descriptor.Descriptor(
  name='YOLOv3DataSource',
  full_name='YOLOv3DataSource',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='label_directory_path', full_name='YOLOv3DataSource.label_directory_path', index=0,
      number=1, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='tfrecords_path', full_name='YOLOv3DataSource.tfrecords_path', index=1,
      number=2, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='image_directory_path', full_name='YOLOv3DataSource.image_directory_path', index=2,
      number=3, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='root_path', full_name='YOLOv3DataSource.root_path', index=3,
      number=4, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='source_weight', full_name='YOLOv3DataSource.source_weight', index=4,
      number=5, type=2, cpp_type=6, label=1,
      has_default_value=False, default_value=float(0),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
  ],
  extensions=[
  ],
  nested_types=[],
  enum_types=[
  ],
  serialized_options=None,
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
    _descriptor.OneofDescriptor(
      name='labels_format', full_name='YOLOv3DataSource.labels_format',
      index=0, containing_type=None, fields=[]),
  ],
  serialized_start=57,
  serialized_end=222,
)


_YOLOV3DATASETCONFIG_TARGETCLASSMAPPINGENTRY = _descriptor.Descriptor(
  name='TargetClassMappingEntry',
  full_name='YOLOv3DatasetConfig.TargetClassMappingEntry',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='key', full_name='YOLOv3DatasetConfig.TargetClassMappingEntry.key', index=0,
      number=1, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='value', full_name='YOLOv3DatasetConfig.TargetClassMappingEntry.value', index=1,
      number=2, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
  ],
  extensions=[
  ],
  nested_types=[],
  enum_types=[
  ],
  serialized_options=_b('8\001'),
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
  ],
  serialized_start=543,
  serialized_end=600,
)

_YOLOV3DATASETCONFIG = _descriptor.Descriptor(
  name='YOLOv3DatasetConfig',
  full_name='YOLOv3DatasetConfig',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='data_sources', full_name='YOLOv3DatasetConfig.data_sources', index=0,
      number=1, type=11, cpp_type=10, label=3,
      has_default_value=False, default_value=[],
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='target_class_mapping', full_name='YOLOv3DatasetConfig.target_class_mapping', index=1,
      number=2, type=11, cpp_type=10, label=3,
      has_default_value=False, default_value=[],
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='validation_fold', full_name='YOLOv3DatasetConfig.validation_fold', index=2,
      number=4, type=13, cpp_type=3, label=1,
      has_default_value=False, default_value=0,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='validation_data_sources', full_name='YOLOv3DatasetConfig.validation_data_sources', index=3,
      number=3, type=11, cpp_type=10, label=3,
      has_default_value=False, default_value=[],
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='include_difficult_in_training', full_name='YOLOv3DatasetConfig.include_difficult_in_training', index=4,
      number=7, type=8, cpp_type=7, label=1,
      has_default_value=False, default_value=False,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='type', full_name='YOLOv3DatasetConfig.type', index=5,
      number=5, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='image_extension', full_name='YOLOv3DatasetConfig.image_extension', index=6,
      number=6, type=9, cpp_type=9, label=1,
      has_default_value=False, default_value=_b("").decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='is_monochrome', full_name='YOLOv3DatasetConfig.is_monochrome', index=7,
      number=8, type=8, cpp_type=7, label=1,
      has_default_value=False, default_value=False,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
  ],
  extensions=[
  ],
  nested_types=[_YOLOV3DATASETCONFIG_TARGETCLASSMAPPINGENTRY, ],
  enum_types=[
  ],
  serialized_options=None,
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
  ],
  serialized_start=225,
  serialized_end=600,
)

_YOLOV3DATASOURCE.oneofs_by_name['labels_format'].fields.append(
  _YOLOV3DATASOURCE.fields_by_name['label_directory_path'])
_YOLOV3DATASOURCE.fields_by_name['label_directory_path'].containing_oneof = _YOLOV3DATASOURCE.oneofs_by_name['labels_format']
_YOLOV3DATASOURCE.oneofs_by_name['labels_format'].fields.append(
  _YOLOV3DATASOURCE.fields_by_name['tfrecords_path'])
_YOLOV3DATASOURCE.fields_by_name['tfrecords_path'].containing_oneof = _YOLOV3DATASOURCE.oneofs_by_name['labels_format']
_YOLOV3DATASETCONFIG_TARGETCLASSMAPPINGENTRY.containing_type = _YOLOV3DATASETCONFIG
_YOLOV3DATASETCONFIG.fields_by_name['data_sources'].message_type = _YOLOV3DATASOURCE
_YOLOV3DATASETCONFIG.fields_by_name['target_class_mapping'].message_type = _YOLOV3DATASETCONFIG_TARGETCLASSMAPPINGENTRY
_YOLOV3DATASETCONFIG.fields_by_name['validation_data_sources'].message_type = _YOLOV3DATASOURCE
DESCRIPTOR.message_types_by_name['YOLOv3DataSource'] = _YOLOV3DATASOURCE
DESCRIPTOR.message_types_by_name['YOLOv3DatasetConfig'] = _YOLOV3DATASETCONFIG
_sym_db.RegisterFileDescriptor(DESCRIPTOR)

YOLOv3DataSource = _reflection.GeneratedProtocolMessageType('YOLOv3DataSource', (_message.Message,), dict(
  DESCRIPTOR = _YOLOV3DATASOURCE,
  __module__ = 'nvidia_tao_tf1.cv.yolo_v3.proto.dataset_config_pb2'
  # @@protoc_insertion_point(class_scope:YOLOv3DataSource)
  ))
_sym_db.RegisterMessage(YOLOv3DataSource)

YOLOv3DatasetConfig = _reflection.GeneratedProtocolMessageType('YOLOv3DatasetConfig', (_message.Message,), dict(

  TargetClassMappingEntry = _reflection.GeneratedProtocolMessageType('TargetClassMappingEntry', (_message.Message,), dict(
    DESCRIPTOR = _YOLOV3DATASETCONFIG_TARGETCLASSMAPPINGENTRY,
    __module__ = 'nvidia_tao_tf1.cv.yolo_v3.proto.dataset_config_pb2'
    # @@protoc_insertion_point(class_scope:YOLOv3DatasetConfig.TargetClassMappingEntry)
    ))
  ,
  DESCRIPTOR = _YOLOV3DATASETCONFIG,
  __module__ = 'nvidia_tao_tf1.cv.yolo_v3.proto.dataset_config_pb2'
  # @@protoc_insertion_point(class_scope:YOLOv3DatasetConfig)
  ))
_sym_db.RegisterMessage(YOLOv3DatasetConfig)
_sym_db.RegisterMessage(YOLOv3DatasetConfig.TargetClassMappingEntry)


_YOLOV3DATASETCONFIG_TARGETCLASSMAPPINGENTRY._options = None
# @@protoc_insertion_point(module_scope)
