denkproto 1.3.0__py3-none-any.whl

denkproto/modelfile_v2_pb2.pyi

@@ -0,0 +1,307 @@
+import validate_pb2 as _validate_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ModelFile(_message.Message):
+    __slots__ = ("protocol_version", "created_at", "file_info", "file_content", "class_labels", "inputs", "outputs")
+    class Version(_message.Message):
+        __slots__ = ("major", "minor", "patch")
+        MAJOR_FIELD_NUMBER: _ClassVar[int]
+        MINOR_FIELD_NUMBER: _ClassVar[int]
+        PATCH_FIELD_NUMBER: _ClassVar[int]
+        major: int
+        minor: int
+        patch: int
+        def __init__(self, major: _Optional[int] = ..., minor: _Optional[int] = ..., patch: _Optional[int] = ...) -> None: ...
+    class Content(_message.Message):
+        __slots__ = ("byte_content", "hash_sha256", "compression_method", "encryption_method", "key_slots")
+        class CompressionMethod(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+            __slots__ = ()
+            CM_NONE: _ClassVar[ModelFile.Content.CompressionMethod]
+        CM_NONE: ModelFile.Content.CompressionMethod
+        class EncryptionMethod(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+            __slots__ = ()
+            EM_NONE: _ClassVar[ModelFile.Content.EncryptionMethod]
+            EM_AES_GCM: _ClassVar[ModelFile.Content.EncryptionMethod]
+        EM_NONE: ModelFile.Content.EncryptionMethod
+        EM_AES_GCM: ModelFile.Content.EncryptionMethod
+        class KeySlot(_message.Message):
+            __slots__ = ("wrapped_key", "wrapping_method")
+            WRAPPED_KEY_FIELD_NUMBER: _ClassVar[int]
+            WRAPPING_METHOD_FIELD_NUMBER: _ClassVar[int]
+            wrapped_key: bytes
+            wrapping_method: ModelFile.Content.EncryptionMethod
+            def __init__(self, wrapped_key: _Optional[bytes] = ..., wrapping_method: _Optional[_Union[ModelFile.Content.EncryptionMethod, str]] = ...) -> None: ...
+        class KeySlotsEntry(_message.Message):
+            __slots__ = ("key", "value")
+            KEY_FIELD_NUMBER: _ClassVar[int]
+            VALUE_FIELD_NUMBER: _ClassVar[int]
+            key: str
+            value: ModelFile.Content.KeySlot
+            def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[ModelFile.Content.KeySlot, _Mapping]] = ...) -> None: ...
+        BYTE_CONTENT_FIELD_NUMBER: _ClassVar[int]
+        HASH_SHA256_FIELD_NUMBER: _ClassVar[int]
+        COMPRESSION_METHOD_FIELD_NUMBER: _ClassVar[int]
+        ENCRYPTION_METHOD_FIELD_NUMBER: _ClassVar[int]
+        KEY_SLOTS_FIELD_NUMBER: _ClassVar[int]
+        byte_content: bytes
+        hash_sha256: bytes
+        compression_method: ModelFile.Content.CompressionMethod
+        encryption_method: ModelFile.Content.EncryptionMethod
+        key_slots: _containers.MessageMap[str, ModelFile.Content.KeySlot]
+        def __init__(self, byte_content: _Optional[bytes] = ..., hash_sha256: _Optional[bytes] = ..., compression_method: _Optional[_Union[ModelFile.Content.CompressionMethod, str]] = ..., encryption_method: _Optional[_Union[ModelFile.Content.EncryptionMethod, str]] = ..., key_slots: _Optional[_Mapping[str, ModelFile.Content.KeySlot]] = ...) -> None: ...
+    class ClassLabel(_message.Message):
+        __slots__ = ("class_label_id", "name", "short_name", "color")
+        CLASS_LABEL_ID_FIELD_NUMBER: _ClassVar[int]
+        NAME_FIELD_NUMBER: _ClassVar[int]
+        SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
+        COLOR_FIELD_NUMBER: _ClassVar[int]
+        class_label_id: str
+        name: str
+        short_name: str
+        color: str
+        def __init__(self, class_label_id: _Optional[str] = ..., name: _Optional[str] = ..., short_name: _Optional[str] = ..., color: _Optional[str] = ...) -> None: ...
+    class ImageSize(_message.Message):
+        __slots__ = ("width", "height", "channels")
+        WIDTH_FIELD_NUMBER: _ClassVar[int]
+        HEIGHT_FIELD_NUMBER: _ClassVar[int]
+        CHANNELS_FIELD_NUMBER: _ClassVar[int]
+        width: int
+        height: int
+        channels: int
+        def __init__(self, width: _Optional[int] = ..., height: _Optional[int] = ..., channels: _Optional[int] = ...) -> None: ...
+    class RegionFromEdge(_message.Message):
+        __slots__ = ("left", "right", "top", "bottom")
+        LEFT_FIELD_NUMBER: _ClassVar[int]
+        RIGHT_FIELD_NUMBER: _ClassVar[int]
+        TOP_FIELD_NUMBER: _ClassVar[int]
+        BOTTOM_FIELD_NUMBER: _ClassVar[int]
+        left: float
+        right: float
+        top: float
+        bottom: float
+        def __init__(self, left: _Optional[float] = ..., right: _Optional[float] = ..., top: _Optional[float] = ..., bottom: _Optional[float] = ...) -> None: ...
+    class Input(_message.Message):
+        __slots__ = ("input_name", "image_format")
+        class ImageInputFormat(_message.Message):
+            __slots__ = ("exact_image_size", "divisible_image_size", "region_of_interest")
+            class ExactImageSizeRequirement(_message.Message):
+                __slots__ = ("image_size",)
+                IMAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+                image_size: ModelFile.ImageSize
+                def __init__(self, image_size: _Optional[_Union[ModelFile.ImageSize, _Mapping]] = ...) -> None: ...
+            class DivisibleImageSizeRequirement(_message.Message):
+                __slots__ = ("image_size_divisors", "minimum_image_size", "suggested_image_size")
+                IMAGE_SIZE_DIVISORS_FIELD_NUMBER: _ClassVar[int]
+                MINIMUM_IMAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+                SUGGESTED_IMAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+                image_size_divisors: ModelFile.ImageSize
+                minimum_image_size: ModelFile.ImageSize
+                suggested_image_size: ModelFile.ImageSize
+                def __init__(self, image_size_divisors: _Optional[_Union[ModelFile.ImageSize, _Mapping]] = ..., minimum_image_size: _Optional[_Union[ModelFile.ImageSize, _Mapping]] = ..., suggested_image_size: _Optional[_Union[ModelFile.ImageSize, _Mapping]] = ...) -> None: ...
+            EXACT_IMAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+            DIVISIBLE_IMAGE_SIZE_FIELD_NUMBER: _ClassVar[int]
+            REGION_OF_INTEREST_FIELD_NUMBER: _ClassVar[int]
+            exact_image_size: ModelFile.Input.ImageInputFormat.ExactImageSizeRequirement
+            divisible_image_size: ModelFile.Input.ImageInputFormat.DivisibleImageSizeRequirement
+            region_of_interest: ModelFile.RegionFromEdge
+            def __init__(self, exact_image_size: _Optional[_Union[ModelFile.Input.ImageInputFormat.ExactImageSizeRequirement, _Mapping]] = ..., divisible_image_size: _Optional[_Union[ModelFile.Input.ImageInputFormat.DivisibleImageSizeRequirement, _Mapping]] = ..., region_of_interest: _Optional[_Union[ModelFile.RegionFromEdge, _Mapping]] = ...) -> None: ...
+        INPUT_NAME_FIELD_NUMBER: _ClassVar[int]
+        IMAGE_FORMAT_FIELD_NUMBER: _ClassVar[int]
+        input_name: str
+        image_format: ModelFile.Input.ImageInputFormat
+        def __init__(self, input_name: _Optional[str] = ..., image_format: _Optional[_Union[ModelFile.Input.ImageInputFormat, _Mapping]] = ...) -> None: ...
+    class Output(_message.Message):
+        __slots__ = ("output_name", "scalar_format", "segmentation_maps_format", "bounding_boxes_format", "bounding_box_segmentations_format", "ocr_format")
+        class ScalarOutputFormat(_message.Message):
+            __slots__ = ()
+            def __init__(self) -> None: ...
+        class SegmentationMapsOutputFormat(_message.Message):
+            __slots__ = ()
+            def __init__(self) -> None: ...
+        class BoundingBoxesOutputFormat(_message.Message):
+            __slots__ = ("x1_offset", "y1_offset", "x2_offset", "y2_offset", "confidence_offset", "class_label_index_offset", "angle_offset")
+            X1_OFFSET_FIELD_NUMBER: _ClassVar[int]
+            Y1_OFFSET_FIELD_NUMBER: _ClassVar[int]
+            X2_OFFSET_FIELD_NUMBER: _ClassVar[int]
+            Y2_OFFSET_FIELD_NUMBER: _ClassVar[int]
+            CONFIDENCE_OFFSET_FIELD_NUMBER: _ClassVar[int]
+            CLASS_LABEL_INDEX_OFFSET_FIELD_NUMBER: _ClassVar[int]
+            ANGLE_OFFSET_FIELD_NUMBER: _ClassVar[int]
+            x1_offset: int
+            y1_offset: int
+            x2_offset: int
+            y2_offset: int
+            confidence_offset: int
+            class_label_index_offset: int
+            angle_offset: int
+            def __init__(self, x1_offset: _Optional[int] = ..., y1_offset: _Optional[int] = ..., x2_offset: _Optional[int] = ..., y2_offset: _Optional[int] = ..., confidence_offset: _Optional[int] = ..., class_label_index_offset: _Optional[int] = ..., angle_offset: _Optional[int] = ...) -> None: ...
+        class BoundingBoxSegmentationsOutputFormat(_message.Message):
+            __slots__ = ("relative_to_bounding_box",)
+            RELATIVE_TO_BOUNDING_BOX_FIELD_NUMBER: _ClassVar[int]
+            relative_to_bounding_box: bool
+            def __init__(self, relative_to_bounding_box: bool = ...) -> None: ...
+        class OcrOutputFormat(_message.Message):
+            __slots__ = ("characters", "character_restrictions")
+            class OcrFormatRestrictionBlock(_message.Message):
+                __slots__ = ("number_of_characters", "allowed_character_indexes")
+                NUMBER_OF_CHARACTERS_FIELD_NUMBER: _ClassVar[int]
+                ALLOWED_CHARACTER_INDEXES_FIELD_NUMBER: _ClassVar[int]
+                number_of_characters: int
+                allowed_character_indexes: _containers.RepeatedScalarFieldContainer[int]
+                def __init__(self, number_of_characters: _Optional[int] = ..., allowed_character_indexes: _Optional[_Iterable[int]] = ...) -> None: ...
+            class Character(_message.Message):
+                __slots__ = ("utf8_representation", "character_type", "ignore")
+                class CharacterType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+                    __slots__ = ()
+                    CT_REGULAR: _ClassVar[ModelFile.Output.OcrOutputFormat.Character.CharacterType]
+                    CT_START_OF_TEXT: _ClassVar[ModelFile.Output.OcrOutputFormat.Character.CharacterType]
+                    CT_END_OF_TEXT: _ClassVar[ModelFile.Output.OcrOutputFormat.Character.CharacterType]
+                    CT_PADDING: _ClassVar[ModelFile.Output.OcrOutputFormat.Character.CharacterType]
+                CT_REGULAR: ModelFile.Output.OcrOutputFormat.Character.CharacterType
+                CT_START_OF_TEXT: ModelFile.Output.OcrOutputFormat.Character.CharacterType
+                CT_END_OF_TEXT: ModelFile.Output.OcrOutputFormat.Character.CharacterType
+                CT_PADDING: ModelFile.Output.OcrOutputFormat.Character.CharacterType
+                UTF8_REPRESENTATION_FIELD_NUMBER: _ClassVar[int]
+                CHARACTER_TYPE_FIELD_NUMBER: _ClassVar[int]
+                IGNORE_FIELD_NUMBER: _ClassVar[int]
+                utf8_representation: bytes
+                character_type: ModelFile.Output.OcrOutputFormat.Character.CharacterType
+                ignore: bool
+                def __init__(self, utf8_representation: _Optional[bytes] = ..., character_type: _Optional[_Union[ModelFile.Output.OcrOutputFormat.Character.CharacterType, str]] = ..., ignore: bool = ...) -> None: ...
+            CHARACTERS_FIELD_NUMBER: _ClassVar[int]
+            CHARACTER_RESTRICTIONS_FIELD_NUMBER: _ClassVar[int]
+            characters: _containers.RepeatedCompositeFieldContainer[ModelFile.Output.OcrOutputFormat.Character]
+            character_restrictions: ModelFile.Output.OcrOutputFormat.OcrFormatRestrictionBlock
+            def __init__(self, characters: _Optional[_Iterable[_Union[ModelFile.Output.OcrOutputFormat.Character, _Mapping]]] = ..., character_restrictions: _Optional[_Union[ModelFile.Output.OcrOutputFormat.OcrFormatRestrictionBlock, _Mapping]] = ...) -> None: ...
+        OUTPUT_NAME_FIELD_NUMBER: _ClassVar[int]
+        SCALAR_FORMAT_FIELD_NUMBER: _ClassVar[int]
+        SEGMENTATION_MAPS_FORMAT_FIELD_NUMBER: _ClassVar[int]
+        BOUNDING_BOXES_FORMAT_FIELD_NUMBER: _ClassVar[int]
+        BOUNDING_BOX_SEGMENTATIONS_FORMAT_FIELD_NUMBER: _ClassVar[int]
+        OCR_FORMAT_FIELD_NUMBER: _ClassVar[int]
+        output_name: str
+        scalar_format: ModelFile.Output.ScalarOutputFormat
+        segmentation_maps_format: ModelFile.Output.SegmentationMapsOutputFormat
+        bounding_boxes_format: ModelFile.Output.BoundingBoxesOutputFormat
+        bounding_box_segmentations_format: ModelFile.Output.BoundingBoxSegmentationsOutputFormat
+        ocr_format: ModelFile.Output.OcrOutputFormat
+        def __init__(self, output_name: _Optional[str] = ..., scalar_format: _Optional[_Union[ModelFile.Output.ScalarOutputFormat, _Mapping]] = ..., segmentation_maps_format: _Optional[_Union[ModelFile.Output.SegmentationMapsOutputFormat, _Mapping]] = ..., bounding_boxes_format: _Optional[_Union[ModelFile.Output.BoundingBoxesOutputFormat, _Mapping]] = ..., bounding_box_segmentations_format: _Optional[_Union[ModelFile.Output.BoundingBoxSegmentationsOutputFormat, _Mapping]] = ..., ocr_format: _Optional[_Union[ModelFile.Output.OcrOutputFormat, _Mapping]] = ...) -> None: ...
+    class FileInfo(_message.Message):
+        __slots__ = ("network_name", "network_id", "network_experiment_id", "network_snapshot_id", "network_type", "network_flavor", "network_version", "runtime_version", "precision", "minimum_libdenkflow_version")
+        class NetworkType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+            __slots__ = ()
+            NT_UNKNOWN: _ClassVar[ModelFile.FileInfo.NetworkType]
+            NT_CLASSIFICATION: _ClassVar[ModelFile.FileInfo.NetworkType]
+            NT_SEGMENTATION: _ClassVar[ModelFile.FileInfo.NetworkType]
+            NT_INSTANCE_SEGMENTATION: _ClassVar[ModelFile.FileInfo.NetworkType]
+            NT_OBJECT_DETECTION: _ClassVar[ModelFile.FileInfo.NetworkType]
+            NT_ANOMALY_DETECTION: _ClassVar[ModelFile.FileInfo.NetworkType]
+            NT_OPTICAL_CHARACTER_RECOGNITION: _ClassVar[ModelFile.FileInfo.NetworkType]
+        NT_UNKNOWN: ModelFile.FileInfo.NetworkType
+        NT_CLASSIFICATION: ModelFile.FileInfo.NetworkType
+        NT_SEGMENTATION: ModelFile.FileInfo.NetworkType
+        NT_INSTANCE_SEGMENTATION: ModelFile.FileInfo.NetworkType
+        NT_OBJECT_DETECTION: ModelFile.FileInfo.NetworkType
+        NT_ANOMALY_DETECTION: ModelFile.FileInfo.NetworkType
+        NT_OPTICAL_CHARACTER_RECOGNITION: ModelFile.FileInfo.NetworkType
+        class Precision(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+            __slots__ = ()
+            P_UNDEFINED: _ClassVar[ModelFile.FileInfo.Precision]
+            P_MIXED_PRECISION: _ClassVar[ModelFile.FileInfo.Precision]
+            P_FP8: _ClassVar[ModelFile.FileInfo.Precision]
+            P_FP16: _ClassVar[ModelFile.FileInfo.Precision]
+            P_FP32: _ClassVar[ModelFile.FileInfo.Precision]
+            P_FP64: _ClassVar[ModelFile.FileInfo.Precision]
+            P_BF8: _ClassVar[ModelFile.FileInfo.Precision]
+            P_BF16: _ClassVar[ModelFile.FileInfo.Precision]
+            P_BF32: _ClassVar[ModelFile.FileInfo.Precision]
+            P_BF64: _ClassVar[ModelFile.FileInfo.Precision]
+            P_INT8: _ClassVar[ModelFile.FileInfo.Precision]
+            P_INT16: _ClassVar[ModelFile.FileInfo.Precision]
+            P_INT32: _ClassVar[ModelFile.FileInfo.Precision]
+            P_INT64: _ClassVar[ModelFile.FileInfo.Precision]
+            P_UINT8: _ClassVar[ModelFile.FileInfo.Precision]
+            P_UINT16: _ClassVar[ModelFile.FileInfo.Precision]
+            P_UINT32: _ClassVar[ModelFile.FileInfo.Precision]
+            P_UINT64: _ClassVar[ModelFile.FileInfo.Precision]
+        P_UNDEFINED: ModelFile.FileInfo.Precision
+        P_MIXED_PRECISION: ModelFile.FileInfo.Precision
+        P_FP8: ModelFile.FileInfo.Precision
+        P_FP16: ModelFile.FileInfo.Precision
+        P_FP32: ModelFile.FileInfo.Precision
+        P_FP64: ModelFile.FileInfo.Precision
+        P_BF8: ModelFile.FileInfo.Precision
+        P_BF16: ModelFile.FileInfo.Precision
+        P_BF32: ModelFile.FileInfo.Precision
+        P_BF64: ModelFile.FileInfo.Precision
+        P_INT8: ModelFile.FileInfo.Precision
+        P_INT16: ModelFile.FileInfo.Precision
+        P_INT32: ModelFile.FileInfo.Precision
+        P_INT64: ModelFile.FileInfo.Precision
+        P_UINT8: ModelFile.FileInfo.Precision
+        P_UINT16: ModelFile.FileInfo.Precision
+        P_UINT32: ModelFile.FileInfo.Precision
+        P_UINT64: ModelFile.FileInfo.Precision
+        NETWORK_NAME_FIELD_NUMBER: _ClassVar[int]
+        NETWORK_ID_FIELD_NUMBER: _ClassVar[int]
+        NETWORK_EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int]
+        NETWORK_SNAPSHOT_ID_FIELD_NUMBER: _ClassVar[int]
+        NETWORK_TYPE_FIELD_NUMBER: _ClassVar[int]
+        NETWORK_FLAVOR_FIELD_NUMBER: _ClassVar[int]
+        NETWORK_VERSION_FIELD_NUMBER: _ClassVar[int]
+        RUNTIME_VERSION_FIELD_NUMBER: _ClassVar[int]
+        PRECISION_FIELD_NUMBER: _ClassVar[int]
+        MINIMUM_LIBDENKFLOW_VERSION_FIELD_NUMBER: _ClassVar[int]
+        network_name: str
+        network_id: str
+        network_experiment_id: str
+        network_snapshot_id: str
+        network_type: ModelFile.FileInfo.NetworkType
+        network_flavor: str
+        network_version: ModelFile.Version
+        runtime_version: ModelFile.Version
+        precision: ModelFile.FileInfo.Precision
+        minimum_libdenkflow_version: ModelFile.Version
+        def __init__(self, network_name: _Optional[str] = ..., network_id: _Optional[str] = ..., network_experiment_id: _Optional[str] = ..., network_snapshot_id: _Optional[str] = ..., network_type: _Optional[_Union[ModelFile.FileInfo.NetworkType, str]] = ..., network_flavor: _Optional[str] = ..., network_version: _Optional[_Union[ModelFile.Version, _Mapping]] = ..., runtime_version: _Optional[_Union[ModelFile.Version, _Mapping]] = ..., precision: _Optional[_Union[ModelFile.FileInfo.Precision, str]] = ..., minimum_libdenkflow_version: _Optional[_Union[ModelFile.Version, _Mapping]] = ...) -> None: ...
+    class FileContent(_message.Message):
+        __slots__ = ("default_model", "tensorrt_model")
+        class DefaultModel(_message.Message):
+            __slots__ = ("model_data",)
+            MODEL_DATA_FIELD_NUMBER: _ClassVar[int]
+            model_data: ModelFile.Content
+            def __init__(self, model_data: _Optional[_Union[ModelFile.Content, _Mapping]] = ...) -> None: ...
+        class TensorRTModel(_message.Message):
+            __slots__ = ("model_data", "calibration_cache", "calibration_flatbuffers")
+            MODEL_DATA_FIELD_NUMBER: _ClassVar[int]
+            CALIBRATION_CACHE_FIELD_NUMBER: _ClassVar[int]
+            CALIBRATION_FLATBUFFERS_FIELD_NUMBER: _ClassVar[int]
+            model_data: ModelFile.Content
+            calibration_cache: ModelFile.Content
+            calibration_flatbuffers: ModelFile.Content
+            def __init__(self, model_data: _Optional[_Union[ModelFile.Content, _Mapping]] = ..., calibration_cache: _Optional[_Union[ModelFile.Content, _Mapping]] = ..., calibration_flatbuffers: _Optional[_Union[ModelFile.Content, _Mapping]] = ...) -> None: ...
+        DEFAULT_MODEL_FIELD_NUMBER: _ClassVar[int]
+        TENSORRT_MODEL_FIELD_NUMBER: _ClassVar[int]
+        default_model: ModelFile.FileContent.DefaultModel
+        tensorrt_model: ModelFile.FileContent.TensorRTModel
+        def __init__(self, default_model: _Optional[_Union[ModelFile.FileContent.DefaultModel, _Mapping]] = ..., tensorrt_model: _Optional[_Union[ModelFile.FileContent.TensorRTModel, _Mapping]] = ...) -> None: ...
+    PROTOCOL_VERSION_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    FILE_INFO_FIELD_NUMBER: _ClassVar[int]
+    FILE_CONTENT_FIELD_NUMBER: _ClassVar[int]
+    CLASS_LABELS_FIELD_NUMBER: _ClassVar[int]
+    INPUTS_FIELD_NUMBER: _ClassVar[int]
+    OUTPUTS_FIELD_NUMBER: _ClassVar[int]
+    protocol_version: ModelFile.Version
+    created_at: int
+    file_info: ModelFile.FileInfo
+    file_content: ModelFile.FileContent
+    class_labels: _containers.RepeatedCompositeFieldContainer[ModelFile.ClassLabel]
+    inputs: _containers.RepeatedCompositeFieldContainer[ModelFile.Input]
+    outputs: _containers.RepeatedCompositeFieldContainer[ModelFile.Output]
+    def __init__(self, protocol_version: _Optional[_Union[ModelFile.Version, _Mapping]] = ..., created_at: _Optional[int] = ..., file_info: _Optional[_Union[ModelFile.FileInfo, _Mapping]] = ..., file_content: _Optional[_Union[ModelFile.FileContent, _Mapping]] = ..., class_labels: _Optional[_Iterable[_Union[ModelFile.ClassLabel, _Mapping]]] = ..., inputs: _Optional[_Iterable[_Union[ModelFile.Input, _Mapping]]] = ..., outputs: _Optional[_Iterable[_Union[ModelFile.Output, _Mapping]]] = ...) -> None: ...

denkproto/modelfile_v2_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.68.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in modelfile_v2_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

denkproto/prediction_pb2.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: prediction.proto
+# Protobuf Python Version: 5.28.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    28,
+    1,
+    '',
+    'prediction.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+import denkproto.geometry_pb2 as geometry__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x10prediction.proto\x12\nprediction\x1a\x0egeometry.proto\"]\n\x18\x43lassificationPrediction\x12\x10\n\x08label_id\x18\x01 \x01(\t\x12\x13\n\x0bprobability\x18\x02 \x01(\x01\x12\x1a\n\x12interpretation_map\x18\x03 \x01(\x0c\"\xb1\x01\n\x19ObjectDetectionPrediction\x12\x10\n\x08label_id\x18\x01 \x01(\t\x12\x11\n\ttop_leftx\x18\x02 \x01(\r\x12\x12\n\ntop_left_y\x18\x03 \x01(\r\x12\r\n\x05width\x18\x04 \x01(\r\x12\x0e\n\x06height\x18\x05 \x01(\r\x12\x13\n\x0bprobability\x18\x06 \x01(\x01\x12\r\n\x05\x61ngle\x18\x07 \x01(\x01\x12\x18\n\x10\x66ull_orientation\x18\x08 \x01(\x08\"}\n\x1eInstanceSegmentationPrediction\x12\x10\n\x08label_id\x18\x01 \x01(\t\x12\x12\n\ntop_left_x\x18\x02 \x01(\r\x12\x12\n\ntop_left_y\x18\x03 \x01(\r\x12\x0c\n\x04mask\x18\x04 \x01(\x0c\x12\x13\n\x0bprobability\x18\x05 \x01(\x01\"=\n\x13\x43haracterPrediction\x12\x11\n\tcharacter\x18\x01 \x01(\t\x12\x13\n\x0bprobability\x18\x02 \x01(\x01\"\xd5\x01\n\rOcrPrediction\x12\x10\n\x08label_id\x18\x01 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\x12>\n\x15\x63haracter_predictions\x18\x03 \x03(\x0b\x32\x1f.prediction.CharacterPrediction\x12-\n\x0c\x62ounding_box\x18\x04 \x01(\x0b\x32\x15.geometry.BoundingBoxH\x00\x12$\n\x07polygon\x18\x05 \x01(\x0b\x32\x11.geometry.PolygonH\x00\x42\x0f\n\rgeometry_data\"\xcd\x02\n\nPrediction\x12\x0e\n\x06height\x18\x01 \x01(\x05\x12\r\n\x05width\x18\x02 \x01(\x05\x12H\n\x1a\x63lassification_predictions\x18\x03 \x03(\x0b\x32$.prediction.ClassificationPrediction\x12K\n\x1cobject_detection_predictions\x18\x04 \x03(\x0b\x32%.prediction.ObjectDetectionPrediction\x12U\n!instance_segmentation_predictions\x18\x05 \x03(\x0b\x32*.prediction.InstanceSegmentationPrediction\x12\x32\n\x0focr_predictions\x18\x06 \x03(\x0b\x32\x19.prediction.OcrPredictionBEZ+github.com/DENKweit/denkproto-go/prediction\xaa\x02\x15\x44\x45NK.Proto.Predictionb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'prediction_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z+github.com/DENKweit/denkproto-go/prediction\252\002\025DENK.Proto.Prediction'
+  _globals['_CLASSIFICATIONPREDICTION']._serialized_start=48
+  _globals['_CLASSIFICATIONPREDICTION']._serialized_end=141
+  _globals['_OBJECTDETECTIONPREDICTION']._serialized_start=144
+  _globals['_OBJECTDETECTIONPREDICTION']._serialized_end=321
+  _globals['_INSTANCESEGMENTATIONPREDICTION']._serialized_start=323
+  _globals['_INSTANCESEGMENTATIONPREDICTION']._serialized_end=448
+  _globals['_CHARACTERPREDICTION']._serialized_start=450
+  _globals['_CHARACTERPREDICTION']._serialized_end=511
+  _globals['_OCRPREDICTION']._serialized_start=514
+  _globals['_OCRPREDICTION']._serialized_end=727
+  _globals['_PREDICTION']._serialized_start=730
+  _globals['_PREDICTION']._serialized_end=1063
+# @@protoc_insertion_point(module_scope)

denkproto/prediction_pb2.pyi

@@ -0,0 +1,89 @@
+import geometry_pb2 as _geometry_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ClassificationPrediction(_message.Message):
+    __slots__ = ("label_id", "probability", "interpretation_map")
+    LABEL_ID_FIELD_NUMBER: _ClassVar[int]
+    PROBABILITY_FIELD_NUMBER: _ClassVar[int]
+    INTERPRETATION_MAP_FIELD_NUMBER: _ClassVar[int]
+    label_id: str
+    probability: float
+    interpretation_map: bytes
+    def __init__(self, label_id: _Optional[str] = ..., probability: _Optional[float] = ..., interpretation_map: _Optional[bytes] = ...) -> None: ...
+
+class ObjectDetectionPrediction(_message.Message):
+    __slots__ = ("label_id", "top_leftx", "top_left_y", "width", "height", "probability", "angle", "full_orientation")
+    LABEL_ID_FIELD_NUMBER: _ClassVar[int]
+    TOP_LEFTX_FIELD_NUMBER: _ClassVar[int]
+    TOP_LEFT_Y_FIELD_NUMBER: _ClassVar[int]
+    WIDTH_FIELD_NUMBER: _ClassVar[int]
+    HEIGHT_FIELD_NUMBER: _ClassVar[int]
+    PROBABILITY_FIELD_NUMBER: _ClassVar[int]
+    ANGLE_FIELD_NUMBER: _ClassVar[int]
+    FULL_ORIENTATION_FIELD_NUMBER: _ClassVar[int]
+    label_id: str
+    top_leftx: int
+    top_left_y: int
+    width: int
+    height: int
+    probability: float
+    angle: float
+    full_orientation: bool
+    def __init__(self, label_id: _Optional[str] = ..., top_leftx: _Optional[int] = ..., top_left_y: _Optional[int] = ..., width: _Optional[int] = ..., height: _Optional[int] = ..., probability: _Optional[float] = ..., angle: _Optional[float] = ..., full_orientation: bool = ...) -> None: ...
+
+class InstanceSegmentationPrediction(_message.Message):
+    __slots__ = ("label_id", "top_left_x", "top_left_y", "mask", "probability")
+    LABEL_ID_FIELD_NUMBER: _ClassVar[int]
+    TOP_LEFT_X_FIELD_NUMBER: _ClassVar[int]
+    TOP_LEFT_Y_FIELD_NUMBER: _ClassVar[int]
+    MASK_FIELD_NUMBER: _ClassVar[int]
+    PROBABILITY_FIELD_NUMBER: _ClassVar[int]
+    label_id: str
+    top_left_x: int
+    top_left_y: int
+    mask: bytes
+    probability: float
+    def __init__(self, label_id: _Optional[str] = ..., top_left_x: _Optional[int] = ..., top_left_y: _Optional[int] = ..., mask: _Optional[bytes] = ..., probability: _Optional[float] = ...) -> None: ...
+
+class CharacterPrediction(_message.Message):
+    __slots__ = ("character", "probability")
+    CHARACTER_FIELD_NUMBER: _ClassVar[int]
+    PROBABILITY_FIELD_NUMBER: _ClassVar[int]
+    character: str
+    probability: float
+    def __init__(self, character: _Optional[str] = ..., probability: _Optional[float] = ...) -> None: ...
+
+class OcrPrediction(_message.Message):
+    __slots__ = ("label_id", "text", "character_predictions", "bounding_box", "polygon")
+    LABEL_ID_FIELD_NUMBER: _ClassVar[int]
+    TEXT_FIELD_NUMBER: _ClassVar[int]
+    CHARACTER_PREDICTIONS_FIELD_NUMBER: _ClassVar[int]
+    BOUNDING_BOX_FIELD_NUMBER: _ClassVar[int]
+    POLYGON_FIELD_NUMBER: _ClassVar[int]
+    label_id: str
+    text: str
+    character_predictions: _containers.RepeatedCompositeFieldContainer[CharacterPrediction]
+    bounding_box: _geometry_pb2.BoundingBox
+    polygon: _geometry_pb2.Polygon
+    def __init__(self, label_id: _Optional[str] = ..., text: _Optional[str] = ..., character_predictions: _Optional[_Iterable[_Union[CharacterPrediction, _Mapping]]] = ..., bounding_box: _Optional[_Union[_geometry_pb2.BoundingBox, _Mapping]] = ..., polygon: _Optional[_Union[_geometry_pb2.Polygon, _Mapping]] = ...) -> None: ...
+
+class Prediction(_message.Message):
+    __slots__ = ("height", "width", "classification_predictions", "object_detection_predictions", "instance_segmentation_predictions", "ocr_predictions")
+    HEIGHT_FIELD_NUMBER: _ClassVar[int]
+    WIDTH_FIELD_NUMBER: _ClassVar[int]
+    CLASSIFICATION_PREDICTIONS_FIELD_NUMBER: _ClassVar[int]
+    OBJECT_DETECTION_PREDICTIONS_FIELD_NUMBER: _ClassVar[int]
+    INSTANCE_SEGMENTATION_PREDICTIONS_FIELD_NUMBER: _ClassVar[int]
+    OCR_PREDICTIONS_FIELD_NUMBER: _ClassVar[int]
+    height: int
+    width: int
+    classification_predictions: _containers.RepeatedCompositeFieldContainer[ClassificationPrediction]
+    object_detection_predictions: _containers.RepeatedCompositeFieldContainer[ObjectDetectionPrediction]
+    instance_segmentation_predictions: _containers.RepeatedCompositeFieldContainer[InstanceSegmentationPrediction]
+    ocr_predictions: _containers.RepeatedCompositeFieldContainer[OcrPrediction]
+    def __init__(self, height: _Optional[int] = ..., width: _Optional[int] = ..., classification_predictions: _Optional[_Iterable[_Union[ClassificationPrediction, _Mapping]]] = ..., object_detection_predictions: _Optional[_Iterable[_Union[ObjectDetectionPrediction, _Mapping]]] = ..., instance_segmentation_predictions: _Optional[_Iterable[_Union[InstanceSegmentationPrediction, _Mapping]]] = ..., ocr_predictions: _Optional[_Iterable[_Union[OcrPrediction, _Mapping]]] = ...) -> None: ...

denkproto/prediction_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.68.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in prediction_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

denkproto/prediction_request_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: prediction_request.proto
+# Protobuf Python Version: 5.28.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    28,
+    1,
+    '',
+    'prediction_request.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+import denkproto.geometry_pb2 as geometry__pb2
+import denkproto.request_pb2 as request__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18prediction_request.proto\x12\x12prediction_request\x1a\x0egeometry.proto\x1a\rrequest.proto\"\xa6\x01\n\tOcrObject\x12\n\n\x02id\x18\x01 \x01(\t\x12\x10\n\x08label_id\x18\x02 \x01(\t\x12\x15\n\raverage_width\x18\x04 \x01(\x01\x12-\n\x0c\x62ounding_box\x18\x05 \x01(\x0b\x32\x15.geometry.BoundingBoxH\x00\x12$\n\x07polygon\x18\x06 \x01(\x0b\x32\x11.geometry.PolygonH\x00\x42\x0f\n\rgeometry_data\"\xf4\x02\n\x11PredictionRequest\x12\x35\n\x0crequest_type\x18\x01 \x01(\x0e\x32\x1f.prediction_request.RequestType\x12\n\n\x02id\x18\x02 \x01(\t\x12\x19\n\x11owned_by_group_id\x18\x03 \x01(\t\x12\x12\n\nhasura_url\x18\x04 \x01(\t\x12\x1a\n\x12\x63reated_by_user_id\x18\x05 \x01(\t\x12\x1b\n\x13prediction_priority\x18\x06 \x01(\x05\x12\x36\n\x12network_experiment\x18\x07 \x01(\x0b\x32\x1a.request.NetworkExperiment\x12\x1d\n\x05image\x18\x08 \x01(\x0b\x32\x0e.request.Image\x12-\n%request_classification_interpretation\x18\t \x01(\x08\x12.\n\x07objects\x18\n \x03(\x0b\x32\x1d.prediction_request.OcrObject*B\n\x0bRequestType\x12\x1c\n\x18REQUEST_TYPE_UNSPECIFIED\x10\x00\x12\x0c\n\x08STANDARD\x10\x01\x12\x07\n\x03OCR\x10\x02\x42TZ3github.com/DENKweit/denkproto-go/prediction_request\xaa\x02\x1c\x44\x45NK.Proto.PredictionRequestb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'prediction_request_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z3github.com/DENKweit/denkproto-go/prediction_request\252\002\034DENK.Proto.PredictionRequest'
+  _globals['_REQUESTTYPE']._serialized_start=623
+  _globals['_REQUESTTYPE']._serialized_end=689
+  _globals['_OCROBJECT']._serialized_start=80
+  _globals['_OCROBJECT']._serialized_end=246
+  _globals['_PREDICTIONREQUEST']._serialized_start=249
+  _globals['_PREDICTIONREQUEST']._serialized_end=621
+# @@protoc_insertion_point(module_scope)

denkproto/prediction_request_pb2.pyi

@@ -0,0 +1,56 @@
+import geometry_pb2 as _geometry_pb2
+import request_pb2 as _request_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class RequestType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    REQUEST_TYPE_UNSPECIFIED: _ClassVar[RequestType]
+    STANDARD: _ClassVar[RequestType]
+    OCR: _ClassVar[RequestType]
+REQUEST_TYPE_UNSPECIFIED: RequestType
+STANDARD: RequestType
+OCR: RequestType
+
+class OcrObject(_message.Message):
+    __slots__ = ("id", "label_id", "average_width", "bounding_box", "polygon")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    LABEL_ID_FIELD_NUMBER: _ClassVar[int]
+    AVERAGE_WIDTH_FIELD_NUMBER: _ClassVar[int]
+    BOUNDING_BOX_FIELD_NUMBER: _ClassVar[int]
+    POLYGON_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    label_id: str
+    average_width: float
+    bounding_box: _geometry_pb2.BoundingBox
+    polygon: _geometry_pb2.Polygon
+    def __init__(self, id: _Optional[str] = ..., label_id: _Optional[str] = ..., average_width: _Optional[float] = ..., bounding_box: _Optional[_Union[_geometry_pb2.BoundingBox, _Mapping]] = ..., polygon: _Optional[_Union[_geometry_pb2.Polygon, _Mapping]] = ...) -> None: ...
+
+class PredictionRequest(_message.Message):
+    __slots__ = ("request_type", "id", "owned_by_group_id", "hasura_url", "created_by_user_id", "prediction_priority", "network_experiment", "image", "request_classification_interpretation", "objects")
+    REQUEST_TYPE_FIELD_NUMBER: _ClassVar[int]
+    ID_FIELD_NUMBER: _ClassVar[int]
+    OWNED_BY_GROUP_ID_FIELD_NUMBER: _ClassVar[int]
+    HASURA_URL_FIELD_NUMBER: _ClassVar[int]
+    CREATED_BY_USER_ID_FIELD_NUMBER: _ClassVar[int]
+    PREDICTION_PRIORITY_FIELD_NUMBER: _ClassVar[int]
+    NETWORK_EXPERIMENT_FIELD_NUMBER: _ClassVar[int]
+    IMAGE_FIELD_NUMBER: _ClassVar[int]
+    REQUEST_CLASSIFICATION_INTERPRETATION_FIELD_NUMBER: _ClassVar[int]
+    OBJECTS_FIELD_NUMBER: _ClassVar[int]
+    request_type: RequestType
+    id: str
+    owned_by_group_id: str
+    hasura_url: str
+    created_by_user_id: str
+    prediction_priority: int
+    network_experiment: _request_pb2.NetworkExperiment
+    image: _request_pb2.Image
+    request_classification_interpretation: bool
+    objects: _containers.RepeatedCompositeFieldContainer[OcrObject]
+    def __init__(self, request_type: _Optional[_Union[RequestType, str]] = ..., id: _Optional[str] = ..., owned_by_group_id: _Optional[str] = ..., hasura_url: _Optional[str] = ..., created_by_user_id: _Optional[str] = ..., prediction_priority: _Optional[int] = ..., network_experiment: _Optional[_Union[_request_pb2.NetworkExperiment, _Mapping]] = ..., image: _Optional[_Union[_request_pb2.Image, _Mapping]] = ..., request_classification_interpretation: bool = ..., objects: _Optional[_Iterable[_Union[OcrObject, _Mapping]]] = ...) -> None: ...

denkproto/prediction_request_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.68.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in prediction_request_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )