File size: 1,552 Bytes
57db94b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
__all__: list[str] = []
import cv2.gapi.onnx.ep
import cv2.typing
import typing as _typing
from cv2.gapi.onnx import ep as ep
# Enumerations
TraitAs_TENSOR: int
TRAIT_AS_TENSOR: int
TraitAs_IMAGE: int
TRAIT_AS_IMAGE: int
TraitAs = int
"""One of [TraitAs_TENSOR, TRAIT_AS_TENSOR, TraitAs_IMAGE, TRAIT_AS_IMAGE]"""
# Classes
class PyParams:
# Functions
@_typing.overload
def __init__(self) -> None: ...
@_typing.overload
def __init__(self, tag: str, model_path: str) -> None: ...
def cfgMeanStd(self, layer_name: str, m: cv2.typing.Scalar, s: cv2.typing.Scalar) -> PyParams: ...
def cfgNormalize(self, layer_name: str, flag: bool) -> PyParams: ...
@_typing.overload
def cfgAddExecutionProvider(self, ep: cv2.gapi.onnx.ep.OpenVINO) -> PyParams: ...
@_typing.overload
def cfgAddExecutionProvider(self, ep: cv2.gapi.onnx.ep.DirectML) -> PyParams: ...
@_typing.overload
def cfgAddExecutionProvider(self, ep: cv2.gapi.onnx.ep.CoreML) -> PyParams: ...
@_typing.overload
def cfgAddExecutionProvider(self, ep: cv2.gapi.onnx.ep.CUDA) -> PyParams: ...
@_typing.overload
def cfgAddExecutionProvider(self, ep: cv2.gapi.onnx.ep.TensorRT) -> PyParams: ...
def cfgDisableMemPattern(self) -> PyParams: ...
def cfgSessionOptions(self, options: cv2.typing.map_string_and_string) -> PyParams: ...
def cfgOptLevel(self, opt_level: int) -> PyParams: ...
# Functions
def params(tag: str, model_path: str) -> PyParams: ...
|