MCPcopy Create free account

hub / github.com/D-Robotics/rdk_model_zoo / functions

Functions523 in github.com/D-Robotics/rdk_model_zoo

Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/efficientformer/runtime/python/efficientformer.py:214
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/mobileone/runtime/python/mobileone.py:214
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in OpenCV BGR format. prompts: Prom
samples/vision/yoloworld/runtime/python/yoloworld_det.py:262
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/fasternet/runtime/python/fasternet.py:214
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/hgnetv2/runtime/python/hgnetv2.py:214
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/efficientnet/runtime/python/efficientnet.py:214
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/fastvit/runtime/python/fastvit.py:214
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/resnext/runtime/python/resnext.py:215
Method__call__
Provides functional-style calling capability.
samples/vision/convnext/runtime/python/convnext.py:220
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/efficientformerv2/runtime/python/efficientformerv2.py:214
Method__call__
Provide functional-style calling capability. Args: img (np.ndarray): Input image in BGR format. Returns:
samples/vision/ultralytics_yolo26/runtime/python/yolo26_cls.py:205
Method__call__
Provide functional-style calling capability. Args: img (np.ndarray): Input image in BGR format. Returns:
samples/vision/ultralytics_yolo26/runtime/python/yolo26_pose.py:304
Method__call__
Provide functional-style calling capability. Args: img (np.ndarray): Input image in BGR format. Returns:
samples/vision/ultralytics_yolo26/runtime/python/yolo26_det.py:276
Method__call__
Provide functional-style calling capability. Args: img (np.ndarray): Input image in BGR format. image_format
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:430
Method__call__
Provide functional-style calling capability. Args: img (np.ndarray): Input image in BGR format. Returns:
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:392
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/repvit/runtime/python/repvit.py:214
Method__call__
Provide function-style inference for one image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_cls.py:162
Method__call__
Provide function-style inference for one image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_det.py:289
Method__call__
Provide function-style inference for one image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_seg.py:284
Method__call__
Provide function-style inference for one image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_pose.py:260
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/mobilenetv4/runtime/python/mobilenetv4.py:214
Method__call__
Provide functional-style access to `predict()`. Args: image: Input image in BGR format. resize_type: Optiona
samples/vision/googlenet/runtime/python/googlenet.py:214
Method__call__
Provide functional-style calling capability, equivalent to ``predict()``. Args: image: Input image in BGR format. Return
samples/vision/modnet/runtime/python/modnet.py:264
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/efficientvit/runtime/python/efficientvit.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/edgenext/runtime/python/edgenext.py:74
Method__init__
Initialize the model and resolve runtime metadata. Args: config (LPRNetConfig): Runtime configuration containing the mod
samples/vision/lprnet/runtime/python/lprnet.py:93
Method__init__
Initialize the model, load metadata, and precompute grids and anchors. Args: config (YOLOv5Config): Configuration object
samples/vision/yolov5/runtime/python/yolov5_det.py:87
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/resnet/runtime/python/resnet.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/repvgg/runtime/python/repvgg.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/mobilenetv3/runtime/python/mobilenetv3.py:74
Method__init__
(self, bpe_path: str = default_bpe())
samples/vision/clip/runtime/python/simple_tokenizer.py:67
Method__init__
Initialize both CLIP encoder runtimes. Args: config: Runtime configuration containing image and text model paths.
samples/vision/clip/runtime/python/clip_retrieval.py:65
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/mobilenetv2/runtime/python/mobilenetv2.py:73
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/vargconvnet/runtime/python/vargconvnet.py:68
Method__init__
Initialize the model and extract runtime metadata. Args: config: Configuration object containing model path and params.
samples/vision/yoloe/runtime/python/yoloe_seg.py:209
Method__init__
(self, alphabet: str, ignore_case: bool = False)
samples/vision/PaddleOCR/runtime/python/paddleocr.py:89
Method__init__
Initialize both detection and recognition models. Args: config (PaddleOCRConfig): Configuration object containing model paths and
samples/vision/PaddleOCR/runtime/python/paddleocr.py:138
Method__init__
( self, model_path: str, input_width: int = 1024, input_height: int = 512,
samples/vision/pp_liteseg/runtime/python/pp_liteseg.py:59
Method__init__
(self, config: PPLiteSegConfig)
samples/vision/pp_liteseg/runtime/python/pp_liteseg.py:79
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/mobilenetv1/runtime/python/mobilenetv1.py:73
Method__init__
Initialize the FCOS runtime wrapper. Args: config (FCOSConfig): Runtime configuration for FCOS inference. The i
samples/vision/fcos/runtime/python/fcos_det.py:85
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/repghost/runtime/python/repghost.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/efficientformer/runtime/python/efficientformer.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/mobileone/runtime/python/mobileone.py:74
Method__init__
Initialize the runtime and load offline vocabulary embeddings. Args: config: Runtime configuration containing model and
samples/vision/yoloworld/runtime/python/yoloworld_det.py:71
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/fasternet/runtime/python/fasternet.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/hgnetv2/runtime/python/hgnetv2.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/efficientnet/runtime/python/efficientnet.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/fastvit/runtime/python/fastvit.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/resnext/runtime/python/resnext.py:74
Method__init__
Initializes the model, loads the HBM, and extracts metadata. Args: config (ConvNeXtConfig): Configuration object contain
samples/vision/convnext/runtime/python/convnext.py:72
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/efficientformerv2/runtime/python/efficientformerv2.py:74
Method__init__
Initialize the model and extract metadata. Args: config (YOLO26ClsConfig): Configuration object containing model path an
samples/vision/ultralytics_yolo26/runtime/python/yolo26_cls.py:71
Method__init__
Initialize the model, load metadata, and precompute decoding grids. Args: config (YOLO26PoseConfig): Configuration objec
samples/vision/ultralytics_yolo26/runtime/python/yolo26_pose.py:78
Method__init__
Initialize the model, load metadata, and precompute decoding grids. Args: config (YOLO26Config): Configuration object co
samples/vision/ultralytics_yolo26/runtime/python/yolo26_det.py:78
Method__init__
Initialize the model and extract runtime metadata. Args: config (YOLO26SegConfig): Configuration object containing model
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:194
Method__init__
Initialize the model, load metadata, and precompute decoding grids. Args: config (YOLO26OBBConfig): Configuration object
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:87
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/repvit/runtime/python/repvit.py:74
Method__init__
Initialize the classification runtime wrapper. Args: config: Classification runtime configuration for the current model.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_cls.py:76
Method__init__
Initialize the runtime wrapper and decoding metadata. Args: config: Detection runtime configuration for the current model.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_det.py:98
Method__init__
Initialize the segmentation runtime wrapper. Args: config: Segmentation runtime configuration for the current model.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_seg.py:84
Method__init__
Initialize the pose runtime wrapper. Args: config: Pose runtime configuration for the current model.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_pose.py:82
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/mobilenetv4/runtime/python/mobilenetv4.py:74
Method__init__
Initialize the runtime wrapper and parse model metadata. Args: config: Runtime configuration including model path and in
samples/vision/googlenet/runtime/python/googlenet.py:74
Method__init__
Initialize the model and extract runtime metadata. Args: config: Configuration object containing model path and params.
samples/vision/modnet/runtime/python/modnet.py:78
Method_obb_sort_key
(item)
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:129
Method_pose_sort_key
(item)
samples/vision/ultralytics_yolo26/runtime/python/yolo26_pose.py:119
Method_seg_sort_key
(item)
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:243
Functionbgr_to_nv12_planes
Convert a BGR image to NV12 format (Y and UV planes). This function converts a BGR image into NV12 format by first transforming it into plana
utils/py_utils/preprocess.py:42
Functionbpu_classify_forward
Modified forward method for YOLO26 Classify Head (Fully Convolutional).
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_cls_bpu.py:45
Functionbpu_detect_forward
Modified forward method for YOLO26 Detect Head (BPU-Optimized). Args: x (List[torch.Tensor]): Input feature maps from the neck. Retu
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_detect_bpu.py:44
Functionbpu_obb_forward
Modified forward method for YOLO26 OBB Head (BPU-Optimized). Args: x (List[torch.Tensor]): Input feature maps from the neck. Returns
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_obb_bpu.py:46
Functionbpu_pose_forward
Modified forward method for YOLO26 Pose Head (BPU-Optimized). Returns: List[torch.Tensor]: 9 tensors ( [Cls, Box, Kpts] * 3 scales ).
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_pose_bpu.py:43
Functionbpu_segment_forward
Modified forward method for YOLO26 Segment Head (BPU-Optimized). Returns: List[torch.Tensor]: 10 tensors ( [Cls, Box, MC] * 3 + Proto ).
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_seg_bpu.py:47
Functioncrop_and_rotate_image
Crop and rotate a region from an image using a rotated bounding box. This function extracts a rotated rectangular region defined by a four-po
utils/py_utils/postprocess.py:759
Functiondecode_boxes
Decode bounding boxes from distributional predictions. This function decodes bounding box coordinates from distribution-based regression outp
utils/py_utils/postprocess.py:414
Functiondecode_kpts
Decode keypoint coordinates from model output. This function decodes keypoint predictions from the model output tensor into pixel coordinates
utils/py_utils/postprocess.py:502
Functiondecode_ltrb_boxes
Decode anchor-relative LTRB regression outputs to `xyxy` boxes. This helper converts YOLO-style left, top, right, and bottom distances predic
utils/py_utils/postprocess.py:318
Functiondecode_masks
Decode instance segmentation masks. This function generates instance-level binary masks by linearly combining mask prototype features with pe
utils/py_utils/postprocess.py:452
Functiondecode_outputs
Decode all feature maps from the model output. This function iterates over all detection heads, reshapes and reorders the raw output tensors,
utils/py_utils/postprocess.py:598
Functiondequantize_outputs
Dequantize a dictionary of quantized model outputs. This function applies tensor dequantization to each model output using its corresponding
utils/py_utils/postprocess.py:147
Functiondownload_model_if_needed
Ensure that the model file exists locally, downloading it if necessary. If the model file specified by `model_path` does not exist, this function
utils/py_utils/file_io.py:49
Functiondraw_classification
Draw Top-K results on image.
utils/py_utils/visualize.py:149
Functiondraw_cls_yolo26
Log YOLO26 classification results (returns image unchanged).
utils/py_utils/visualize.py:236
Functiondraw_detect_yolo26
Draw YOLO26 detection results from `(boxes, scores, cls_ids)` arrays.
utils/py_utils/visualize.py:164
Functiondraw_masks
Overlay semi-transparent instance masks.
utils/py_utils/visualize.py:78
Functiondraw_obb_yolo26
Draw YOLO26 OBB results and log details.
utils/py_utils/visualize.py:207
Functiondraw_pose_yolo26
Draw YOLO26 pose results and log details.
utils/py_utils/visualize.py:222
Functiondraw_seg_yolo26
Draw YOLO26 segmentation results and log details.
utils/py_utils/visualize.py:171
Functionfilter_classification
Filter classification outputs using a raw confidence threshold. This function selects classification predictions whose maximum logit value ex
utils/py_utils/postprocess.py:289
Functionfilter_mces
Extract MCES features for selected predictions. This function selects MCES feature vectors corresponding to the given valid prediction indice
utils/py_utils/postprocess.py:342
Functionfilter_predictions
Filter detection predictions by confidence threshold. This function combines objectness scores and class probabilities to compute final confi
utils/py_utils/postprocess.py:362
Functionget_bounding_boxes
Extract minimum-area bounding boxes from polygon contours. This function computes the minimum-area bounding rectangle for each polygon contou
utils/py_utils/postprocess.py:634
Functionget_soc_name
Get the SoC (System-on-Chip) name of the current device. The SoC name is read from the system file `/sys/class/boardinfo/soc_name`. If the fi
utils/py_utils/inspect.py:46
Functionload_class_names
Load class names from a label file. Each line in the file is treated as a single class name. Empty lines are ignored. Args: path
utils/py_utils/file_io.py:113
Functionload_image
Load an image from a file path using OpenCV. The image is loaded in BGR color format, which is the default format used by OpenCV. Args:
utils/py_utils/file_io.py:92
Functionload_imagenet_labels
Load ImageNet labels from a file. This function attempts to load labels that might be stored as a Python dictionary string (e.g., "{0: 'label
utils/py_utils/file_io.py:132
Functionmain
samples/vision/yolov5/runtime/cpp/main.cc:107
Functionmain
samples/vision/ultralytics_yolo/runtime/cpp/classify/main.cc:471
Functionmain
samples/vision/ultralytics_yolo/runtime/cpp/segment/main.cc:303
← previousnext →401–500 of 523, ranked by callers