Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a image classification model exported by torchvision.ResNet.

:param model_file: (str)Path of model file, e.g resnet/resnet50.onnx
:param params_file: (str)Path of parameters file, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model, default is ONNX
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model, default is ONNX
"""

# call super() to initialize the backend_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -89,9 +89,9 @@ def __init__(
"""Load a YOLOv5Cls model exported by YOLOv5Cls.

:param model_file: (str)Path of model file, e.g ./YOLOv5Cls.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""

super(YOLOv5Cls, self).__init__(runtime_option)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -88,10 +88,10 @@ def __init__(
"""Load a image classification model exported by PaddleClas.

:param model_file: (str)Path of model file, e.g resnet50/inference.pdmodel
:param params_file: (str)Path of parameters file, e.g resnet50/inference.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g resnet50/inference.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param config_file: (str) Path of configuration file for deploy, e.g resnet50/inference_cls.yaml
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""

super(PaddleClasModel, self).__init__(runtime_option)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -85,10 +85,10 @@ def __init__(
"""Load a image PPShiTuV2Recognizer model exported by PaddleClas.

:param model_file: (str)Path of model file, e.g PPLCNet/inference.pdmodel
:param params_file: (str)Path of parameters file, e.g PPLCNet/inference.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g PPLCNet/inference.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param config_file: (str) Path of configuration file for deploy, e.g PPLCNet/inference_cls.yaml
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""

super(PPShiTuV2Recognizer, self).__init__(runtime_option)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -105,9 +105,9 @@ def __init__(
"""Load a FastestDet model exported by FastestDet.

:param model_file: (str)Path of model file, e.g ./FastestDet.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""

super(FastestDet, self).__init__(runtime_option)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a NanoDetPlus model exported by NanoDet.

:param model_file: (str)Path of model file, e.g ./nanodet.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -148,9 +148,9 @@ def __init__(self, model_file, runtime_option=None, model_format=ModelFormat.RKN
"""Load a RKYOLOV5 model exported by RKYOLOV5.

:param model_file: (str)Path of model file, e.g ./yolov5.rknn
:param params_file: (str)Path of parameters file, e.g , if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g , if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down Expand Up @@ -207,7 +207,7 @@ def __init__(self, model_file, runtime_option=None, model_format=ModelFormat.RKN

:param model_file: (str)Path of model file, e.g ./yolox.rknn
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down Expand Up @@ -264,7 +264,7 @@ def __init__(self, model_file, runtime_option=None, model_format=ModelFormat.RKN

:param model_file: (str)Path of model file, e.g ./yolov7.rknn
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a ScaledYOLOv4 model exported by ScaledYOLOv4.

:param model_file: (str)Path of model file, e.g ./scaled_yolov4.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a YOLOR model exported by YOLOR

:param model_file: (str)Path of model file, e.g ./yolor.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -174,9 +174,9 @@ def __init__(
"""Load a YOLOv5 model exported by YOLOv5.

:param model_file: (str)Path of model file, e.g ./yolov5.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a YOLOv5Lite model exported by YOLOv5Lite.

:param model_file: (str)Path of model file, e.g ./yolov5lite.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -174,9 +174,9 @@ def __init__(
"""Load a YOLOv5Seg model exported by YOLOv5.

:param model_file: (str)Path of model file, e.g ./yolov5s-seg.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
super(YOLOv5Seg, self).__init__(runtime_option)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a YOLOv6 model exported by YOLOv6.

:param model_file: (str)Path of model file, e.g ./yolov6.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -134,9 +134,9 @@ def __init__(
"""Load a YOLOv7 model exported by YOLOv7.

:param model_file: (str)Path of model file, e.g ./yolov7.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a YOLOv7End2EndORT model exported by YOLOv7.

:param model_file: (str)Path of model file, e.g ./yolov7end2end_ort.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a YOLOv7End2EndTRT model exported by YOLOv7.

:param model_file: (str)Path of model file, e.g ./yolov7end2end_trt.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -174,9 +174,9 @@ def __init__(
"""Load a YOLOv8 model exported by YOLOv8.

:param model_file: (str)Path of model file, e.g ./yolov8s.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
super(YOLOv8, self).__init__(runtime_option)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@ def __init__(
"""Load a YOLOX model exported by YOLOX.

:param model_file: (str)Path of model file, e.g ./yolox.onnx
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param params_file: (str)Path of parameters file, e.g yolox/model.pdiparams, if the model_format is ModelFormat.ONNX, this param will be ignored, can be set as empty string
:param runtime_option: (ultra_infer.RuntimeOption)RuntimeOption for inference this model, if it's None, will use the default backend on CPU
:param model_format: (ultra_infer.ModelForamt)Model format of the loaded model
:param model_format: (ultra_infer.ModelFormat)Model format of the loaded model
"""
# 调用基函数进行backend_option的初始化
# 初始化后的option保存在self._runtime_option
Expand Down
Loading
Loading