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
↓ 1 callers
Function
draw_boxes_and_texts
Draw detected boxes on the image and recognized texts on a white panel. Args: image (np.ndarray): Original image. boxes (list): L
samples/vision/PaddleOCR/runtime/python/main.py:67
↓ 1 callers
Function
draw_detection_results
Log and draw detection results from `(boxes, scores, cls_ids)` arrays.
utils/py_utils/visualize.py:67
↓ 1 callers
Function
draw_detections
* @brief Draw detection results on image * @param img Image to draw on * @param detections Detection results * @param x_scale X scale factor for co
samples/vision/ultralytics_yolo/runtime/cpp/detect/main.cc:274
↓ 1 callers
Function
draw_legend
(canvas, cls_ids, palette, names)
samples/vision/pp_liteseg/evaluator/infer_board.py:79
↓ 1 callers
Method
draw_legend
Overlay a small class legend on the top-right corner of canvas.
samples/vision/pp_liteseg/runtime/python/pp_liteseg.py:159
↓ 1 callers
Function
draw_pose
Draw pose estimation results.
utils/py_utils/visualize.py:118
↓ 1 callers
Function
draw_pose
* @brief Draw pose detection results */
samples/vision/ultralytics_yolo/runtime/cpp/pose/main.cc:280
↓ 1 callers
Function
draw_results
Draw YOLOWorld detection results on the original image. Args: image: Original BGR image. boxes: Detection boxes in original
samples/vision/yoloworld/runtime/python/main.py:91
↓ 1 callers
Function
draw_rotated_boxes
Draw rotated bounding boxes (OBB).
utils/py_utils/visualize.py:100
↓ 1 callers
Function
draw_scores
Draw CLIP similarity scores on the input image. Args: image: Original BGR image. texts: Candidate text prompts. scor
samples/vision/clip/runtime/python/main.py:90
↓ 1 callers
Method
encode
(self, text)
samples/vision/clip/runtime/python/simple_tokenizer.py:125
↓ 1 callers
Method
encode_text
Execute the ONNX text encoder for a list of prompts. Args: texts: Prompt list to encode. Returns: T
samples/vision/clip/runtime/python/clip_retrieval.py:170
↓ 1 callers
Function
export_bpu_onnx
Export YOLO26 model to BPU-friendly ONNX.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_detect_bpu.py:76
↓ 1 callers
Function
export_cls_bpu
Export YOLO26 Classification model.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_cls_bpu.py:71
↓ 1 callers
Function
export_obb_bpu
Export YOLO26 OBB model. Args: model_path (str): Path to input .pt file. output_name (str): Path to save output .onnx file.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_obb_bpu.py:85
↓ 1 callers
Function
export_pose_bpu
Export YOLO26 Pose model.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_pose_bpu.py:79
↓ 1 callers
Function
export_seg_bpu
Export YOLO26 Segmentation model.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_seg_bpu.py:79
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/efficientvit/runtime/python/efficientvit.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/edgenext/runtime/python/edgenext.py:144
↓ 1 callers
Method
forward
Run BPU inference and return the raw logits tensor. Args: input_tensors (Dict[str, Dict[str, np.ndarray]]): Runtime inpu
samples/vision/lprnet/runtime/python/lprnet.py:148
↓ 1 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: inputs (Dict[str, Dict[str, np.ndarray]]): Prepared input tensors.
samples/vision/yolov5/runtime/python/yolov5_det.py:182
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/resnet/runtime/python/resnet.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/repvgg/runtime/python/repvgg.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/mobilenetv3/runtime/python/mobilenetv3.py:144
↓ 1 callers
Method
forward
Execute the image encoder on BPU. Args: inputs: Prepared image encoder inputs. Returns: Image featu
samples/vision/clip/runtime/python/clip_retrieval.py:156
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/mobilenetv2/runtime/python/mobilenetv2.py:143
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/vargconvnet/runtime/python/vargconvnet.py:138
↓ 1 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: input_tensor: Prepared input tensors. Returns: Raw output
samples/vision/yoloe/runtime/python/yoloe_seg.py:288
↓ 1 callers
Method
forward
Execute detection model inference on BPU. Args: input_tensor (Dict[str, Dict[str, np.ndarray]]): Prepared input tensors.
samples/vision/PaddleOCR/runtime/python/paddleocr.py:212
↓ 1 callers
Method
forward
Run BPU inference. Args: nv12: Preprocessed NV12 array. Returns: Raw output array of shape (1, H, W, 1) int3
samples/vision/pp_liteseg/runtime/python/pp_liteseg.py:115
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/mobilenetv1/runtime/python/mobilenetv1.py:143
↓ 1 callers
Method
forward
Execute inference on BPU using `hbm_runtime`. Args: inputs (Dict[str, Dict[str, np.ndarray]]): Runtime input dictionary
samples/vision/fcos/runtime/python/fcos_det.py:218
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/repghost/runtime/python/repghost.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/efficientformer/runtime/python/efficientformer.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/mobileone/runtime/python/mobileone.py:144
↓ 1 callers
Method
forward
Execute one BPU inference pass. Args: inputs: Prepared image and text embedding tensors. Returns: R
samples/vision/yoloworld/runtime/python/yoloworld_det.py:164
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/fasternet/runtime/python/fasternet.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/hgnetv2/runtime/python/hgnetv2.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/efficientnet/runtime/python/efficientnet.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/fastvit/runtime/python/fastvit.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/resnext/runtime/python/resnext.py:144
↓ 1 callers
Method
forward
Executes inference on BPU using hbm_runtime. Args: inputs (Dict): Prepared input tensors from pre_process(). Re
samples/vision/convnext/runtime/python/convnext.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/efficientformerv2/runtime/python/efficientformerv2.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/repvit/runtime/python/repvit.py:144
↓ 1 callers
Method
forward
Run BPU inference for one input image. Args: input_tensor: Input tensor dictionary returned by `pre_process`. Returns:
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_det.py:168
↓ 1 callers
Method
forward
Run BPU inference for one input image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_seg.py:154
↓ 1 callers
Method
forward
Run BPU inference for one input image.
samples/vision/ultralytics_yolo/runtime/python/ultralytics_yolo_pose.py:139
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/mobilenetv4/runtime/python/mobilenetv4.py:144
↓ 1 callers
Method
forward
Execute one forward pass on BPU. Args: inputs: Prepared input tensors returned by `pre_process()`. Returns:
samples/vision/googlenet/runtime/python/googlenet.py:144
↓ 1 callers
Method
forward
Execute inference on BPU using hbm_runtime. Args: input_tensor: Prepared input tensors. Returns: Raw output
samples/vision/modnet/runtime/python/modnet.py:202
↓ 1 callers
Function
get_perf_data
(model_file, thread_num)
utils/tools/batch_perf/batch_perf.py:63
↓ 1 callers
Function
get_topk_results
* @brief Get top K classification results */
samples/vision/ultralytics_yolo/runtime/cpp/classify/main.cc:441
↓ 1 callers
Function
load_ground_truth_csv
从 CSV 文件加载 ground truth,返回 {相对路径字符串: 标签} 的字典。 第一列保持原样(不做任何 basename 提取),第二列为标签(0‑based)。 自动跳过标题行(如果第一行包含 'image' 和 'category')。
samples/vision/hgnetv2/evaluator/eval.py:52
↓ 1 callers
Function
main
()
utils/tools/generate_calibration_data/generate_calibration_data.py:39
↓ 1 callers
Function
main
()
utils/tools/batch_perf/batch_perf.py:11
↓ 1 callers
Function
main
()
utils/tools/batch_mapper/batch_mapper.py:33
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_pycocotools.py:26
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_pytorch_generate_labels_seg.py:39
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_batch_cpp.py:35
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_pycocotools_seg.py:26
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_batch_pytorch_generate_labels_seg.py:34
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_pytorch_generate_labels.py:39
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_pycocotools_pose.py:26
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_batch_python.py:35
↓ 1 callers
Function
main
()
utils/tools/batch_eval_pycocotools/eval_pytorch_generate_labels_pose.py:39
↓ 1 callers
Function
main
Run the complete EfficientViT classification pipeline on a single image. The entry follows the standardized sample pattern used in this repo
samples/vision/efficientvit/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete EdgeNeXt classification pipeline on a single image. The entry follows the standardized sample pattern used in this reposito
samples/vision/edgenext/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete LPRNet recognition pipeline on one binary input tensor. The entry follows the standardized sample pattern used in this repo
samples/vision/lprnet/runtime/python/main.py:33
↓ 1 callers
Function
main
Run the complete YOLOv5 detection pipeline on a single image. This function parses command-line arguments, initializes the runtime wrapp
samples/vision/yolov5/runtime/python/main.py:77
↓ 1 callers
Function
main
Run the complete ResNet classification pipeline on a single image. The entry follows the standardized sample pattern used in this repository
samples/vision/resnet/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete RepVGG classification pipeline on a single image. The entry follows the standardized sample pattern used in this repository
samples/vision/repvgg/runtime/python/main.py:67
↓ 1 callers
Function
main
Run the complete MobileNetV3 classification pipeline on a single image. The entry follows the standardized sample pattern used in this repos
samples/vision/mobilenetv3/runtime/python/main.py:68
↓ 1 callers
Function
main
Run CLIP image-text matching on one image. The entry follows the standard sample pattern: parse arguments, construct the runtime wrapper
samples/vision/clip/runtime/python/main.py:112
↓ 1 callers
Function
main
Run the complete MobileNetV2 classification pipeline on a single image. The entry follows the standardized sample pattern used in this repos
samples/vision/mobilenetv2/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete VargConvNet classification pipeline on a single image. The entry follows the standardized sample pattern used in this repos
samples/vision/vargconvnet/runtime/python/main.py:67
↓ 1 callers
Function
main
()
samples/vision/yoloe/conversion/onnx_export/export_yoloe11seg_bpu.py:62
↓ 1 callers
Function
main
Run the complete YOLOE segmentation pipeline on a single image.
samples/vision/yoloe/runtime/python/main.py:74
↓ 1 callers
Function
main
Run the complete PaddleOCR pipeline on a single image. The entry follows the standardized sample pattern used in this repository: 1. Parse de
samples/vision/PaddleOCR/runtime/python/main.py:112
↓ 1 callers
Function
main
()
samples/vision/pp_liteseg/conversion/prepare_calibration.py:25
↓ 1 callers
Function
main
()
samples/vision/pp_liteseg/evaluator/infer_board.py:95
↓ 1 callers
Function
main
Run PP-LiteSeg-STDC1 inference on a single image.
samples/vision/pp_liteseg/runtime/python/main.py:76
↓ 1 callers
Function
main
Run the complete MobileNetV1 classification pipeline on a single image. The entry follows the standardized sample pattern used in this repos
samples/vision/mobilenetv1/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete FCOS detection pipeline on a single image. The entry follows the standardized sample pattern used in this repository: 1
samples/vision/fcos/runtime/python/main.py:76
↓ 1 callers
Function
main
Run the complete RepGhost classification pipeline on a single image. The entry follows the standardized sample pattern used in this reposito
samples/vision/repghost/runtime/python/main.py:67
↓ 1 callers
Function
main
Run the complete EfficientFormer classification pipeline on a single image. The entry follows the standardized sample pattern used in this r
samples/vision/efficientformer/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete MobileOne classification pipeline on a single image. The entry follows the standardized sample pattern used in this reposit
samples/vision/mobileone/runtime/python/main.py:67
↓ 1 callers
Function
main
Run YOLOWorld open-vocabulary detection on one image. The entry follows the standard sample pattern: parse arguments, construct the wrap
samples/vision/yoloworld/runtime/python/main.py:123
↓ 1 callers
Function
main
Run the complete FasterNet classification pipeline on a single image. The entry follows the standardized sample pattern used in this reposit
samples/vision/fasternet/runtime/python/main.py:67
↓ 1 callers
Function
main
()
samples/vision/hgnetv2/evaluator/eval.py:109
↓ 1 callers
Function
main
Run the complete HGNetV2 classification pipeline on a single image. The entry follows the standardized sample pattern used in this repositor
samples/vision/hgnetv2/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete EfficientNet classification pipeline on a single image. The entry follows the standardized sample pattern used in this repo
samples/vision/efficientnet/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete FastViT classification pipeline on a single image. The entry follows the standardized sample pattern used in this repositor
samples/vision/fastvit/runtime/python/main.py:68
↓ 1 callers
Function
main
Run the complete ResNeXt classification pipeline on a single image. The entry follows the standardized sample pattern used in this repositor
samples/vision/resnext/runtime/python/main.py:68
↓ 1 callers
Function
main
Run ConvNeXt inference on a single image. This function orchestrates the complete inference process: - Argument parsing - Model initi
samples/vision/convnext/runtime/python/main.py:63
↓ 1 callers
Function
main
Run the complete EfficientFormerV2 classification pipeline on one image. The entry follows the standardized sample pattern used in this repo
samples/vision/efficientformerv2/runtime/python/main.py:67
↓ 1 callers
Function
main
Main execution block for model conversion.
samples/vision/ultralytics_yolo26/conversion/mapper.py:89
↓ 1 callers
Function
main
Main entry point for detection model export.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_detect_bpu.py:32
↓ 1 callers
Function
main
Main entry point for OBB export.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_obb_bpu.py:34
↓ 1 callers
Function
main
Main entry point for segmentation model export.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_seg_bpu.py:35
↓ 1 callers
Function
main
Main entry point for classification model export.
samples/vision/ultralytics_yolo26/conversion/onnx_export/export_yolo26_cls_bpu.py:33
← previous
next →
101–200 of 523, ranked by callers