Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/D-Robotics/rdk_model_zoo
/ functions
Functions
523 in github.com/D-Robotics/rdk_model_zoo
⨍
Functions
523
◇
Types & classes
84
↓ 22 callers
Function
softmax
* @brief Softmax function for DFL */
samples/vision/ultralytics_yolo/runtime/cpp/pose/main.cc:258
↓ 6 callers
Method
predict
Run the complete OBB inference pipeline on a single image. This method orchestrates the full workflow: - Pre-process the inp
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:372
↓ 5 callers
Method
forward
Run BPU inference for one input image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_cls.py:131
↓ 5 callers
Method
predict
Run the full classification pipeline on one image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_cls.py:156
↓ 5 callers
Method
set_scheduling_params
Set BPU scheduling parameters like priority and core affinity. Args: priority (Optional[int]): Scheduling priority in ra
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:149
↓ 4 callers
Function
draw_detection
* @brief Draw detection results on image */
samples/vision/ultralytics_yolo/runtime/cpp/segment/main.cc:276
↓ 4 callers
Function
resolve_path
Resolve a path to an absolute path. Args: path: The path string to resolve. base_dir: Optional base directory for relative resolu
samples/vision/ultralytics_yolo26/conversion/mapper.py:70
↓ 4 callers
Function
resolve_path
解析路径,支持相对路径和绝对路径
samples/vision/ultralytics_yolo/conversion/mapper.py:55
↓ 4 callers
Function
save_image
Save an image to the target path. Args: path (str): Output image path. image (np.ndarray): Image array to save. Raises:
samples/vision/ultralytics_yolo26/runtime/python/main.py:79
↓ 4 callers
Method
set_scheduling_params
Set optional BPU scheduling parameters.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_cls.py:101
↓ 3 callers
Function
letterbox_resize_gray
Resize a grayscale image using letterbox (aspect-ratio preserving) strategy. This function resizes a grayscale image while preserving its aspect
utils/py_utils/preprocess.py:153
↓ 3 callers
Method
predict
Run the complete HGNetV2 inference pipeline on one image. Args: image: Input image in BGR format. resize_typ
samples/vision/hgnetv2/runtime/python/hgnetv2.py:180
↓ 3 callers
Function
save_image
Save an image to the target path. Args: path: Output image path. image: Image array to save. Raises: RuntimeError: I
samples/vision/ultralytics_yolo/runtime/python/main.py:80
↓ 3 callers
Function
write_index
Write a minimal index.rst with a toctree.
docs/source_reference/sphinx/tools/gen_samples_nav.py:31
↓ 2 callers
Function
bytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
samples/vision/clip/runtime/python/simple_tokenizer.py:16
↓ 2 callers
Method
decode
Decode a sequence of CTC indices into a string. Args: t (np.ndarray): Encoded indices. length (np.ndarray): Length of
samples/vision/PaddleOCR/runtime/python/paddleocr.py:98
↓ 2 callers
Function
draw_boxes
Draw bounding boxes with class names and scores.
utils/py_utils/visualize.py:51
↓ 2 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: input_tensor (Dict[str, Dict[str, np.ndarray]]): Prepared input tensor
samples/vision/ultralytics_yolo26/runtime/python/yolo26_cls.py:147
↓ 2 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: input_tensor (Dict[str, Dict[str, np.ndarray]]): Prepared input tensor
samples/vision/ultralytics_yolo26/runtime/python/yolo26_pose.py:189
↓ 2 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: input_tensor (Dict[str, Dict[str, np.ndarray]]): Prepared input tensor
samples/vision/ultralytics_yolo26/runtime/python/yolo26_det.py:162
↓ 2 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: input_tensor (Dict[str, Dict[str, np.ndarray]]): Prepared input tensor
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:311
↓ 2 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: input_tensor (Dict[str, Dict[str, np.ndarray]]): Prepared input tensor
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:199
↓ 2 callers
Function
gen_anchor
Generate anchor center positions on a square grid. This function generates evenly spaced anchor center coordinates for a square grid of size
utils/py_utils/postprocess.py:395
↓ 2 callers
Function
get_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
samples/vision/clip/runtime/python/simple_tokenizer.py:42
↓ 2 callers
Function
linear2conv
Replace a Linear layer with an equivalent 1x1 Conv2d for BPU.
samples/vision/yoloe/conversion/onnx_export/export_yoloe11seg_bpu.py:34
↓ 2 callers
Method
post_process
Convert raw output tensor to Top-K classification results. Args: outputs (Dict[str, Dict[str, np.ndarray]]): Raw model o
samples/vision/ultralytics_yolo26/runtime/python/yolo26_cls.py:162
↓ 2 callers
Method
post_process
Convert raw outputs to final pose estimation results. Args: outputs (Dict[str, Dict[str, np.ndarray]]): Raw model output
samples/vision/ultralytics_yolo26/runtime/python/yolo26_pose.py:204
↓ 2 callers
Method
post_process
Convert raw model outputs to final detection results. Args: outputs (Dict[str, Dict[str, np.ndarray]]): Raw model output
samples/vision/ultralytics_yolo26/runtime/python/yolo26_det.py:177
↓ 2 callers
Method
post_process
Convert raw model outputs to final segmentation results. Args: outputs (Dict[str, Dict[str, np.ndarray]]): Raw model out
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:326
↓ 2 callers
Method
post_process
Convert raw outputs to final rotated box results. Args: outputs (Dict[str, Dict[str, np.ndarray]]): Raw model outputs.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:304
↓ 2 callers
Method
pre_process
Convert a BGR image to packed NV12 input for hbm_runtime. Args: img (np.ndarray): Input image in BGR format.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_cls.py:115
↓ 2 callers
Method
pre_process
Convert a BGR image to packed NV12 input for hbm_runtime. Args: img (np.ndarray): Input image in BGR format.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_pose.py:157
↓ 2 callers
Method
pre_process
Convert a BGR image to packed NV12 input for hbm_runtime. Args: img (np.ndarray): Input image in BGR format.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_det.py:129
↓ 2 callers
Method
pre_process
Convert a BGR image to packed NV12 input for hbm_runtime. Args: img (np.ndarray): Input image in BGR format.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:277
↓ 2 callers
Method
pre_process
Convert a BGR image to packed NV12 input for hbm_runtime. Args: img (np.ndarray): Input image in BGR format.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:167
↓ 2 callers
Method
predict
Run the complete EfficientViT inference pipeline on one image. Args: image: Input image in BGR format. resiz
samples/vision/efficientvit/runtime/python/efficientvit.py:180
↓ 2 callers
Method
predict
Run the complete EdgeNeXt inference pipeline on one image. Args: image: Input image in BGR format. resize_ty
samples/vision/edgenext/runtime/python/edgenext.py:180
↓ 2 callers
Method
predict
Run the complete LPRNet pipeline on one binary input tensor. Returns: str: Final decoded license plate string produced b
samples/vision/lprnet/runtime/python/lprnet.py:176
↓ 2 callers
Method
predict
High-level interface for one-click inference. Args: image (np.ndarray): Input image in BGR format. Returns:
samples/vision/yolov5/runtime/python/yolov5_det.py:277
↓ 2 callers
Method
predict
Run the complete ResNet inference pipeline on one image. Args: image: Input image in BGR format. resize_type
samples/vision/resnet/runtime/python/resnet.py:181
↓ 2 callers
Method
predict
Run the complete RepVGG inference pipeline on one image. Args: image: Input image in BGR format. resize_type
samples/vision/repvgg/runtime/python/repvgg.py:180
↓ 2 callers
Method
predict
Run the complete MobileNetV3 inference pipeline on one image. Args: image: Input image in BGR format. resize
samples/vision/mobilenetv3/runtime/python/mobilenetv3.py:180
↓ 2 callers
Method
predict
Run the complete CLIP image-text matching pipeline. Args: image: Input image in OpenCV BGR format. texts: Ca
samples/vision/clip/runtime/python/clip_retrieval.py:200
↓ 2 callers
Method
predict
Run the complete MobileNetV2 inference pipeline on one image. Args: image: Input image in BGR format. resize
samples/vision/mobilenetv2/runtime/python/mobilenetv2.py:179
↓ 2 callers
Method
predict
Run the complete VargConvNet inference pipeline on one image. Args: image: Input image in BGR format. resize
samples/vision/vargconvnet/runtime/python/vargconvnet.py:174
↓ 2 callers
Method
predict
Run the complete segmentation pipeline on a single image. Args: img: Input image in BGR format. image_format: Input i
samples/vision/yoloe/runtime/python/yoloe_seg.py:386
↓ 2 callers
Method
predict
Run the complete OCR pipeline: detect → crop → recognize. Args: image (np.ndarray): Input image in BGR format. Returns:
samples/vision/PaddleOCR/runtime/python/paddleocr.py:363
↓ 2 callers
Method
predict
Run the complete MobileNetV1 inference pipeline on one image. Args: image: Input image in BGR format. resize
samples/vision/mobilenetv1/runtime/python/mobilenetv1.py:179
↓ 2 callers
Method
predict
Execute the full FCOS pipeline on one image. Args: image (np.ndarray): Input BGR image. Returns: Tu
samples/vision/fcos/runtime/python/fcos_det.py:332
↓ 2 callers
Method
predict
Run the complete RepGhost inference pipeline on one image. Args: image: Input image in BGR format. resize_ty
samples/vision/repghost/runtime/python/repghost.py:180
↓ 2 callers
Method
predict
Run the complete EfficientFormer inference pipeline on one image. Args: image: Input image in BGR format. re
samples/vision/efficientformer/runtime/python/efficientformer.py:180
↓ 2 callers
Method
predict
Run the complete MobileOne inference pipeline on one image. Args: image: Input image in BGR format. resize_t
samples/vision/mobileone/runtime/python/mobileone.py:180
↓ 2 callers
Method
predict
Run the complete YOLOWorld detection pipeline on one image. Args: image: Input image in OpenCV BGR format. p
samples/vision/yoloworld/runtime/python/yoloworld_det.py:234
↓ 2 callers
Method
predict
Run the complete FasterNet inference pipeline on one image. Args: image: Input image in BGR format. resize_t
samples/vision/fasternet/runtime/python/fasternet.py:180
↓ 2 callers
Method
predict
Run the complete EfficientNet inference pipeline on one image. Args: image: Input image in BGR format. resiz
samples/vision/efficientnet/runtime/python/efficientnet.py:180
↓ 2 callers
Method
predict
Run the complete FastViT inference pipeline on one image. Args: image: Input image in BGR format. resize_typ
samples/vision/fastvit/runtime/python/fastvit.py:180
↓ 2 callers
Method
predict
Run the complete ResNeXt inference pipeline on one image. Args: image: Input image in BGR format. resize_typ
samples/vision/resnext/runtime/python/resnext.py:181
↓ 2 callers
Method
predict
High-level interface for one-click inference. Orchestrates the pipeline: pre_process -> forward -> post_process. Args:
samples/vision/convnext/runtime/python/convnext.py:185
↓ 2 callers
Method
predict
Run the complete EfficientFormerV2 inference pipeline on one image. Args: image: Input image in BGR format.
samples/vision/efficientformerv2/runtime/python/efficientformerv2.py:180
↓ 2 callers
Method
predict
Run the complete RepViT inference pipeline on one image. Args: image: Input image in BGR format. resize_type
samples/vision/repvit/runtime/python/repvit.py:180
↓ 2 callers
Method
predict
Run the complete MobileNetV4 inference pipeline on one image. Args: image: Input image in BGR format. resize
samples/vision/mobilenetv4/runtime/python/mobilenetv4.py:180
↓ 2 callers
Method
predict
Run the complete GoogLeNet inference pipeline on one image. Args: image: Input image in BGR format. resize_t
samples/vision/googlenet/runtime/python/googlenet.py:180
↓ 2 callers
Method
predict
Run the complete matting pipeline on a single image. Args: image: Input image in BGR format. Returns: Alpha
samples/vision/modnet/runtime/python/modnet.py:248
↓ 2 callers
Function
save_image
Save the result image to disk.
samples/vision/modnet/runtime/python/main.py:82
↓ 2 callers
Method
set_scheduling_params
Set optional runtime scheduling parameters. Args: priority: Scheduling priority in the range `0~255`. bpu_co
samples/vision/hgnetv2/runtime/python/hgnetv2.py:93
↓ 2 callers
Function
sigmoid
Apply the sigmoid activation function element-wise. Args: x: Input tensor in logit space. Returns: Tensor converted to proba
utils/py_utils/postprocess.py:44
↓ 1 callers
Method
_build_text_embeddings
Build the fixed 32-slot text embedding tensor for selected prompts. Args: prompts: Prompt words that must exist in the o
samples/vision/yoloworld/runtime/python/yoloworld_det.py:112
↓ 1 callers
Method
_crop_and_rotate
Crop and perspective-rectify a text region from the image. Args: img (np.ndarray): Original image. box (np.ndarray):
samples/vision/PaddleOCR/runtime/python/paddleocr.py:333
↓ 1 callers
Method
_dilate_contours
Dilate contours using the ratio_prime parameter.
samples/vision/PaddleOCR/runtime/python/paddleocr.py:252
↓ 1 callers
Method
_get_bounding_boxes
Compute minimum-area bounding rectangles from dilated polygons.
samples/vision/PaddleOCR/runtime/python/paddleocr.py:271
↓ 1 callers
Method
_nms_rotated
Perform class-wise rotated NMS. Args: rrects: Rotated rectangles in `(cx, cy, w, h, angle_rad)` format. scor
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:237
↓ 1 callers
Method
_rec_forward
Execute recognition model inference on BPU. Args: input_tensor (np.ndarray): Float32 NCHW RGB tensor. Returns:
samples/vision/PaddleOCR/runtime/python/paddleocr.py:299
↓ 1 callers
Method
_rec_post_process
Run CTC decoding on recognition model output. Args: outputs (Dict[str, Dict[str, np.ndarray]]): Raw output tensors. Retu
samples/vision/PaddleOCR/runtime/python/paddleocr.py:310
↓ 1 callers
Method
_rec_pre_process
Pre-process a cropped image for the recognition model. Args: cropped_img (np.ndarray): Cropped text region in BGR. Retur
samples/vision/PaddleOCR/runtime/python/paddleocr.py:284
↓ 1 callers
Method
_resize_with_padding
Resize image with aspect-ratio-preserving scale and symmetric zero-padding. Args: image: Input image. target_size: Ta
samples/vision/modnet/runtime/python/modnet.py:125
↓ 1 callers
Method
_resolve_output_order
Resolve the FCOS output order based on the original fixed protocol. Returns: Tuple[List[str], List[str], List[str]]: Cla
samples/vision/fcos/runtime/python/fcos_det.py:123
↓ 1 callers
Method
_rotated_iou
Compute IoU between two rotated rectangles. Args: a: First rotated rectangle in `(cx, cy, w, h, angle_rad)` format.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:214
↓ 1 callers
Method
_scale_rrect_to_original
Map a rotated rectangle from input scale back to original image scale. Args: rrect (List[float]): Rotated rectangle on m
samples/vision/ultralytics_yolo26/runtime/python/yolo26_obb.py:266
↓ 1 callers
Function
_underline
(title: str, ch: str = "=")
docs/source_reference/sphinx/tools/gen_samples_nav.py:27
↓ 1 callers
Function
basic_clean
(text)
samples/vision/clip/runtime/python/simple_tokenizer.py:54
↓ 1 callers
Function
bgr2nv12
* @brief Convert BGR image to NV12 format */
samples/vision/ultralytics_yolo/runtime/cpp/classify/main.cc:322
↓ 1 callers
Function
bgr2nv12
* @brief Convert BGR image to NV12 format */
samples/vision/ultralytics_yolo/runtime/cpp/segment/main.cc:159
↓ 1 callers
Function
bgr2nv12
* @brief Convert BGR image to NV12 format * @param bgr_img Input BGR image * @return NV12 formatted image (height * 3/2, width, CV_8UC1) */
samples/vision/ultralytics_yolo/runtime/cpp/detect/main.cc:165
↓ 1 callers
Function
bgr2nv12
* @brief Convert BGR image to NV12 format */
samples/vision/ultralytics_yolo/runtime/cpp/pose/main.cc:163
↓ 1 callers
Method
bpe
(self, token)
samples/vision/clip/runtime/python/simple_tokenizer.py:84
↓ 1 callers
Function
collect_images
(src)
samples/vision/pp_liteseg/conversion/prepare_calibration.py:20
↓ 1 callers
Function
collect_images_with_relative_paths
递归遍历 image_root 目录,收集所有图像文件。 返回列表,每个元素为 (相对路径字符串, 绝对路径)。 相对路径使用 '/' 作为分隔符,并且不包含前导 './'。
samples/vision/hgnetv2/evaluator/eval.py:82
↓ 1 callers
Function
colorize
(seg, palette)
samples/vision/pp_liteseg/evaluator/infer_board.py:71
↓ 1 callers
Method
colorize
Map class indices to BGR colors using the Cityscapes palette.
samples/vision/pp_liteseg/runtime/python/pp_liteseg.py:151
↓ 1 callers
Function
composite
Composite the foreground (image) onto a background using the alpha matte. Args: image: Original image in BGR format. matte: Alpha
samples/vision/modnet/runtime/python/main.py:65
↓ 1 callers
Function
convert_linear_to_conv
Replace the final Linear layer with a Conv2d 1x1 layer.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_cls_bpu.py:56
↓ 1 callers
Function
crop_mask
Crop instance masks using their corresponding bounding boxes. Args: masks: Predicted masks with shape ``(N, H, W)``. boxes: Bound
samples/vision/yoloe/runtime/python/yoloe_seg.py:61
↓ 1 callers
Function
crop_mask
Crop instance masks using their corresponding bounding boxes. Args: masks (np.ndarray): Predicted masks with shape `(N, H, W)`.
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:54
↓ 1 callers
Method
decode
(self, tokens)
samples/vision/clip/runtime/python/simple_tokenizer.py:133
↓ 1 callers
Function
decode_layer
Decode a single feature layer from the detection head. This function decodes the raw output tensor of one detection layer into bounding box p
utils/py_utils/postprocess.py:544
↓ 1 callers
Function
decode_plate
Decode the raw model output into the final license plate string. Args: pred_data (np.ndarray): Raw model output after squeezing the
samples/vision/lprnet/runtime/python/lprnet.py:38
↓ 1 callers
Function
decode_seg_layer
Decode one segmentation head output. Args: box_feat (np.ndarray): Bounding box branch output tensor. cls_feat (np.ndarray):
samples/vision/ultralytics_yolo26/runtime/python/yolo26_seg.py:109
↓ 1 callers
Function
decode_seg_layer_dfl
Decode one segmentation head output with DFL box regression. Unlike ``decode_seg_layer`` (which assumes pre-decoded 4-value bbox), this funct
samples/vision/yoloe/runtime/python/yoloe_seg.py:113
↓ 1 callers
Function
default_bpe
()
samples/vision/clip/runtime/python/simple_tokenizer.py:11
↓ 1 callers
Function
dequantize_tensor
Dequantize a quantized tensor to floating-point values. This function converts a quantized tensor (e.g., int8 or uint8) into floating-point v
utils/py_utils/postprocess.py:108
next →
1–100 of 523, ranked by callers