ONNX Node Catalog
Generated category
ONNX Node Catalog
Generated from 28 catalog nodes in AI/ML/ONNX.
Nodes in this category
Showing 28 of 28 generated node docs.
Extract Keypoint
AI/ML/ONNXExtract a specific keypoint from a pose by index or name
Feature Extraction
AI/ML/ONNXExtract feature vectors from images using ONNX models
Feature Similarity
AI/ML/ONNXCompare two feature vectors using cosine similarity or L2 distance
Image Classification
AI/ML/ONNXImage Classification with ONNX-Models. Download models from: MobileNetV2 (https://github.com/onnx/models/tree/main/validated/vision/classification/mobilenet), SqueezeNet (https://github.com/onnx/models/tree/main/validated/vision/classification/squeezenet), ResNet (https://github.com/onnx/models/tree/main/validated/vision/classification/resnet), EfficientNet (https://github.com/onnx/models/tree/main/validated/vision/classification/efficientnet-lite4)
Load ONNX
AI/ML/ONNXLoad ONNX Model from Path
Model Info
AI/ML/ONNXGet ONNX model metadata (inputs, outputs, shapes)
Object Detection
AI/ML/ONNXObject Detection in Images with ONNX-Models. Download models from: TinyYOLOv2 (https://github.com/onnx/models/tree/main/validated/vision/object_detection_segmentation/tiny-yolov2), YOLO (https://github.com/onnx/models/tree/main/validated/vision/object_detection_segmentation), SSD-MobileNet (https://github.com/onnx/models/tree/main/validated/vision/object_detection_segmentation/ssd-mobilenetv1)
Pose Estimation
AI/ML/ONNXDetect human poses and keypoints using ONNX models. Download models from: YOLOv8-Pose (https://docs.ultralytics.com/models/yolov8/), MoveNet (https://tfhub.dev/google/movenet/), HRNet (https://github.com/OAID/TengineKit)
Semantic Segmentation
AI/ML/ONNXSegment images into semantic classes using ONNX models. Download models from: DeepLabV3 (https://github.com/onnx/models/tree/main/validated/vision/object_detection_segmentation/duc), FCN (https://github.com/onnx/models/tree/main/validated/vision/object_detection_segmentation/fcn)
Session Info
AI/ML/ONNXGet information about a loaded ONNX session
Unload ONNX
AI/ML/ONNXRelease ONNX model from cache to free memory
Audio to Mel Spectrogram
AI/ML/ONNX/AudioConvert audio to mel spectrogram for speech models
Load Audio
AI/ML/ONNX/AudioLoad audio file for processing
Resample Audio
AI/ML/ONNX/AudioResample audio to target sample rate
Trim Audio
AI/ML/ONNX/AudioTrim audio to speech segments from VAD
Voice Activity Detection
AI/ML/ONNX/AudioDetect speech segments in audio. Download Silero VAD model from: https://github.com/snakers4/silero-vad/raw/master/src/silero_vad/data/silero_vad.onnx
Batch Image Inference
AI/ML/ONNX/BatchRun ONNX inference on multiple images in batches
Compare Faces
AI/ML/ONNX/FaceCompare two face embeddings for similarity
Crop Faces
AI/ML/ONNX/FaceCrop detected faces from image
Face Detection
AI/ML/ONNX/FaceDetect faces in images. Download models from: UltraFace (https://github.com/onnx/models/tree/main/validated/vision/body_analysis/ultraface), RetinaFace (https://huggingface.co/arnabdhar/retinaface-onnx), SCRFD (https://huggingface.co/onnx-community/scrfd_10g_bnkps)
Face Embedding
AI/ML/ONNX/FaceExtract face embedding for recognition. Download models from: ArcFace (https://huggingface.co/onnx-community/arcface_torch/tree/main), FaceNet (https://huggingface.co/rocca/facenet-onnx)
Named Entity Recognition
AI/ML/ONNX/NLPExtract named entities (persons, organizations, locations, dates, etc.) from text using ONNX models. Supports BERT, RoBERTa, and other transformer-based NER models with automatic tokenization. Download models from: BERT-base-NER (https://huggingface.co/dslim/bert-base-NER), Multilingual NER (https://huggingface.co/Davlan/bert-base-multilingual-cased-ner-hrl), spaCy NER (https://huggingface.co/spacy). Download tokenizer.json from the same model repository.
Crop Text Regions
AI/ML/ONNX/OCRCrop detected text regions from image for recognition
Text Detection
AI/ML/ONNX/OCRDetect text regions in images. Download models from: CRAFT (https://huggingface.co/quocanh34/craft_text_detection_onnx), DBNet (https://huggingface.co/Xenova/dbnet_resnet50_onnx), EAST (https://www.dropbox.com/s/r2ingd0l3zt8hxs/frozen_east_text_detection.tar.gz)
Text Recognition
AI/ML/ONNX/OCRRecognize text from cropped text regions. Download models from: CRNN (https://huggingface.co/Xenova/crnn_onnx), TrOCR (https://huggingface.co/microsoft/trocr-base-printed), PaddleOCR (https://huggingface.co/aapot/paddleocr-onnx)
Colorize Depth
AI/ML/ONNX/VisionConvert depth map to rainbow-colored visualization
Depth Estimation
AI/ML/ONNX/VisionEstimate depth from a single image using ONNX models. Download models from: MiDaS (https://github.com/isl-org/MiDaS/releases), DPT (https://huggingface.co/Intel/dpt-large/tree/main), Depth Anything (https://huggingface.co/depth-anything/Depth-Anything-V2-Small/tree/main)
Depth to Point Cloud
AI/ML/ONNX/VisionConvert depth map to 3D point cloud coordinates