initial commit
This commit is contained in:
51
venv/lib/python3.11/site-packages/cv2/gapi/onnx/__init__.pyi
Normal file
51
venv/lib/python3.11/site-packages/cv2/gapi/onnx/__init__.pyi
Normal file
@@ -0,0 +1,51 @@
|
||||
__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: ...
|
||||
|
||||
|
||||
|
||||
# Functions
|
||||
def params(tag: str, model_path: str) -> PyParams: ...
|
||||
|
||||
|
||||
@@ -0,0 +1,63 @@
|
||||
__all__: list[str] = []
|
||||
|
||||
import cv2.typing
|
||||
import typing as _typing
|
||||
|
||||
|
||||
# Classes
|
||||
class CoreML:
|
||||
# Functions
|
||||
def __init__(self) -> None: ...
|
||||
|
||||
def cfgUseCPUOnly(self) -> CoreML: ...
|
||||
|
||||
def cfgEnableOnSubgraph(self) -> CoreML: ...
|
||||
|
||||
def cfgEnableOnlyNeuralEngine(self) -> CoreML: ...
|
||||
|
||||
|
||||
class CUDA:
|
||||
# Functions
|
||||
@_typing.overload
|
||||
def __init__(self) -> None: ...
|
||||
@_typing.overload
|
||||
def __init__(self, dev_id: int) -> None: ...
|
||||
|
||||
|
||||
class TensorRT:
|
||||
# Functions
|
||||
@_typing.overload
|
||||
def __init__(self) -> None: ...
|
||||
@_typing.overload
|
||||
def __init__(self, dev_id: int) -> None: ...
|
||||
|
||||
|
||||
class OpenVINO:
|
||||
# Functions
|
||||
@_typing.overload
|
||||
def __init__(self) -> None: ...
|
||||
@_typing.overload
|
||||
def __init__(self, dev_type: str) -> None: ...
|
||||
@_typing.overload
|
||||
def __init__(self, params: cv2.typing.map_string_and_string) -> None: ...
|
||||
|
||||
def cfgCacheDir(self, dir: str) -> OpenVINO: ...
|
||||
|
||||
def cfgNumThreads(self, nthreads: int) -> OpenVINO: ...
|
||||
|
||||
def cfgEnableOpenCLThrottling(self) -> OpenVINO: ...
|
||||
|
||||
def cfgEnableDynamicShapes(self) -> OpenVINO: ...
|
||||
|
||||
|
||||
class DirectML:
|
||||
# Functions
|
||||
@_typing.overload
|
||||
def __init__(self) -> None: ...
|
||||
@_typing.overload
|
||||
def __init__(self, device_id: int) -> None: ...
|
||||
@_typing.overload
|
||||
def __init__(self, adapter_name: str) -> None: ...
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user