# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: nvidia_tao_deploy/cv/unet/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()


from nvidia_tao_deploy.cv.unet.proto import data_class_config_pb2 as nvidia__tao__deploy_dot_cv_dot_unet_dot_proto_dot_data__class__config__pb2
from nvidia_tao_deploy.cv.unet.proto import augmentation_config_pb2 as nvidia__tao__deploy_dot_cv_dot_unet_dot_proto_dot_augmentation__config__pb2


DESCRIPTOR = _descriptor.FileDescriptor(
  name='nvidia_tao_deploy/cv/unet/proto/dataset_config.proto',
  package='',
  syntax='proto3',
  serialized_options=None,
  serialized_pb=_b('\n4nvidia_tao_deploy/cv/unet/proto/dataset_config.proto\x1a\x37nvidia_tao_deploy/cv/unet/proto/data_class_config.proto\x1a\x39nvidia_tao_deploy/cv/unet/proto/augmentation_config.proto\"4\n\nDataSource\x12\x12\n\nimage_path\x18\x01 \x01(\t\x12\x12\n\nmasks_path\x18\x02 \x01(\t\"3\n\x0fTrainDataSource\x12 \n\x0b\x64\x61ta_source\x18\x01 \x03(\x0b\x32\x0b.DataSource\"1\n\rValDataSource\x12 \n\x0b\x64\x61ta_source\x18\x01 \x03(\x0b\x32\x0b.DataSource\"2\n\x0eTestDataSource\x12 \n\x0b\x64\x61ta_source\x18\x01 \x03(\x0b\x32\x0b.DataSource\"\xb3\x04\n\rDatasetConfig\x12\x0f\n\x07\x61ugment\x18\x03 \x01(\x08\x12\x13\n\x0b\x66ilter_data\x18\x1f \x01(\x08\x12\x0f\n\x07\x64\x61taset\x18\n \x01(\t\x12\x12\n\ndataloader\x18\x14 \x01(\t\x12\x12\n\npreprocess\x18\x19 \x01(\t\x12\x16\n\x0eresize_padding\x18\x1d \x01(\x08\x12\x15\n\rresize_method\x18\x1e \x01(\t\x12\x18\n\x10input_image_type\x18\x0b \x01(\t\x12,\n\x12train_data_sources\x18\x01 \x01(\x0b\x32\x10.TrainDataSource\x12(\n\x10val_data_sources\x18\x02 \x01(\x0b\x32\x0e.ValDataSource\x12*\n\x11test_data_sources\x18\x04 \x01(\x0b\x32\x0f.TestDataSource\x12+\n\x11\x64\x61ta_class_config\x18\x12 \x01(\x0b\x32\x10.DataClassConfig\x12\x30\n\x13\x61ugmentation_config\x18\x1c \x01(\x0b\x32\x13.AugmentationConfig\x12\x19\n\x11train_images_path\x18\x0c \x01(\t\x12\x18\n\x10train_masks_path\x18\r \x01(\t\x12\x17\n\x0fval_images_path\x18\x0e \x01(\t\x12\x16\n\x0eval_masks_path\x18\x0f \x01(\t\x12\x18\n\x10test_images_path\x18\x10 \x01(\t\x12\x17\n\x0ftest_masks_path\x18\x11 \x01(\tb\x06proto3')
  ,
  dependencies=[nvidia__tao__deploy_dot_cv_dot_unet_dot_proto_dot_data__class__config__pb2.DESCRIPTOR,nvidia__tao__deploy_dot_cv_dot_unet_dot_proto_dot_augmentation__config__pb2.DESCRIPTOR,])




_DATASOURCE = _descriptor.Descriptor(
  name='DataSource',
  full_name='DataSource',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='image_path', full_name='DataSource.image_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='masks_path', full_name='DataSource.masks_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),
  ],
  extensions=[
  ],
  nested_types=[],
  enum_types=[
  ],
  serialized_options=None,
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
  ],
  serialized_start=172,
  serialized_end=224,
)


_TRAINDATASOURCE = _descriptor.Descriptor(
  name='TrainDataSource',
  full_name='TrainDataSource',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='data_source', full_name='TrainDataSource.data_source', 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),
  ],
  extensions=[
  ],
  nested_types=[],
  enum_types=[
  ],
  serialized_options=None,
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
  ],
  serialized_start=226,
  serialized_end=277,
)


_VALDATASOURCE = _descriptor.Descriptor(
  name='ValDataSource',
  full_name='ValDataSource',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='data_source', full_name='ValDataSource.data_source', 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),
  ],
  extensions=[
  ],
  nested_types=[],
  enum_types=[
  ],
  serialized_options=None,
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
  ],
  serialized_start=279,
  serialized_end=328,
)


_TESTDATASOURCE = _descriptor.Descriptor(
  name='TestDataSource',
  full_name='TestDataSource',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='data_source', full_name='TestDataSource.data_source', 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),
  ],
  extensions=[
  ],
  nested_types=[],
  enum_types=[
  ],
  serialized_options=None,
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
  ],
  serialized_start=330,
  serialized_end=380,
)


_DATASETCONFIG = _descriptor.Descriptor(
  name='DatasetConfig',
  full_name='DatasetConfig',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  fields=[
    _descriptor.FieldDescriptor(
      name='augment', full_name='DatasetConfig.augment', index=0,
      number=3, 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='filter_data', full_name='DatasetConfig.filter_data', index=1,
      number=31, 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='dataset', full_name='DatasetConfig.dataset', index=2,
      number=10, 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='dataloader', full_name='DatasetConfig.dataloader', index=3,
      number=20, 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='preprocess', full_name='DatasetConfig.preprocess', index=4,
      number=25, 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='resize_padding', full_name='DatasetConfig.resize_padding', index=5,
      number=29, 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='resize_method', full_name='DatasetConfig.resize_method', index=6,
      number=30, 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='input_image_type', full_name='DatasetConfig.input_image_type', index=7,
      number=11, 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='train_data_sources', full_name='DatasetConfig.train_data_sources', index=8,
      number=1, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='val_data_sources', full_name='DatasetConfig.val_data_sources', index=9,
      number=2, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='test_data_sources', full_name='DatasetConfig.test_data_sources', index=10,
      number=4, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='data_class_config', full_name='DatasetConfig.data_class_config', index=11,
      number=18, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='augmentation_config', full_name='DatasetConfig.augmentation_config', index=12,
      number=28, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR),
    _descriptor.FieldDescriptor(
      name='train_images_path', full_name='DatasetConfig.train_images_path', index=13,
      number=12, 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='train_masks_path', full_name='DatasetConfig.train_masks_path', index=14,
      number=13, 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='val_images_path', full_name='DatasetConfig.val_images_path', index=15,
      number=14, 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='val_masks_path', full_name='DatasetConfig.val_masks_path', index=16,
      number=15, 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='test_images_path', full_name='DatasetConfig.test_images_path', index=17,
      number=16, 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='test_masks_path', full_name='DatasetConfig.test_masks_path', index=18,
      number=17, 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=None,
  is_extendable=False,
  syntax='proto3',
  extension_ranges=[],
  oneofs=[
  ],
  serialized_start=383,
  serialized_end=946,
)

_TRAINDATASOURCE.fields_by_name['data_source'].message_type = _DATASOURCE
_VALDATASOURCE.fields_by_name['data_source'].message_type = _DATASOURCE
_TESTDATASOURCE.fields_by_name['data_source'].message_type = _DATASOURCE
_DATASETCONFIG.fields_by_name['train_data_sources'].message_type = _TRAINDATASOURCE
_DATASETCONFIG.fields_by_name['val_data_sources'].message_type = _VALDATASOURCE
_DATASETCONFIG.fields_by_name['test_data_sources'].message_type = _TESTDATASOURCE
_DATASETCONFIG.fields_by_name['data_class_config'].message_type = nvidia__tao__deploy_dot_cv_dot_unet_dot_proto_dot_data__class__config__pb2._DATACLASSCONFIG
_DATASETCONFIG.fields_by_name['augmentation_config'].message_type = nvidia__tao__deploy_dot_cv_dot_unet_dot_proto_dot_augmentation__config__pb2._AUGMENTATIONCONFIG
DESCRIPTOR.message_types_by_name['DataSource'] = _DATASOURCE
DESCRIPTOR.message_types_by_name['TrainDataSource'] = _TRAINDATASOURCE
DESCRIPTOR.message_types_by_name['ValDataSource'] = _VALDATASOURCE
DESCRIPTOR.message_types_by_name['TestDataSource'] = _TESTDATASOURCE
DESCRIPTOR.message_types_by_name['DatasetConfig'] = _DATASETCONFIG
_sym_db.RegisterFileDescriptor(DESCRIPTOR)

DataSource = _reflection.GeneratedProtocolMessageType('DataSource', (_message.Message,), dict(
  DESCRIPTOR = _DATASOURCE,
  __module__ = 'nvidia_tao_deploy.cv.unet.proto.dataset_config_pb2'
  # @@protoc_insertion_point(class_scope:DataSource)
  ))
_sym_db.RegisterMessage(DataSource)

TrainDataSource = _reflection.GeneratedProtocolMessageType('TrainDataSource', (_message.Message,), dict(
  DESCRIPTOR = _TRAINDATASOURCE,
  __module__ = 'nvidia_tao_deploy.cv.unet.proto.dataset_config_pb2'
  # @@protoc_insertion_point(class_scope:TrainDataSource)
  ))
_sym_db.RegisterMessage(TrainDataSource)

ValDataSource = _reflection.GeneratedProtocolMessageType('ValDataSource', (_message.Message,), dict(
  DESCRIPTOR = _VALDATASOURCE,
  __module__ = 'nvidia_tao_deploy.cv.unet.proto.dataset_config_pb2'
  # @@protoc_insertion_point(class_scope:ValDataSource)
  ))
_sym_db.RegisterMessage(ValDataSource)

TestDataSource = _reflection.GeneratedProtocolMessageType('TestDataSource', (_message.Message,), dict(
  DESCRIPTOR = _TESTDATASOURCE,
  __module__ = 'nvidia_tao_deploy.cv.unet.proto.dataset_config_pb2'
  # @@protoc_insertion_point(class_scope:TestDataSource)
  ))
_sym_db.RegisterMessage(TestDataSource)

DatasetConfig = _reflection.GeneratedProtocolMessageType('DatasetConfig', (_message.Message,), dict(
  DESCRIPTOR = _DATASETCONFIG,
  __module__ = 'nvidia_tao_deploy.cv.unet.proto.dataset_config_pb2'
  # @@protoc_insertion_point(class_scope:DatasetConfig)
  ))
_sym_db.RegisterMessage(DatasetConfig)


# @@protoc_insertion_point(module_scope)
