nvidia-cv-detection

Benchmarks object detection model training performance on NVIDIA GPUs using mmdetection.

7|1|Updated Apr 17, 2026
One-click install
npx skills add https://github.com/DeepLink-org/DeepEval-Skills --skill nvidia-cv-detection-deeplink-org
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: nvidia-cv-detection
Source: https://github.com/DeepLink-org/DeepEval-Skills/tree/main/skills/NVIDIA/cv/nvidia-cv-detection
Command: npx skills add https://github.com/DeepLink-org/DeepEval-Skills --skill nvidia-cv-detection-deeplink-org

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes scripts (resource) components.

What problem does it solve? Measuring training performance of object detection models on NVIDIA GPUs requires manual container setup, distributed training configuration, and log parsing. This Skill automates the full workflow: launching a Docker container, running mmdetection training benchmarks across models and precisions, and extracting AVG_ITER_TIME metrics into a structured JSON result. ## Core Features & Use Cases - Automated Benchmark Execution: Runs batch training benchmarks for faster_rcnn, mask_rcnn, cascade_rcnn, retinanet, yolov3, fcos, ssd300, centernet, solo, and swin_mask_rcnn on COCO with fixed fp16/fp32 precisions. - Structured Metric Collection: Parses MMEngine training logs for AVG_ITER_TIME, DATA, and OP metrics and writes all model/GPU/precision combinations into a single eval_result.json. - Flexible GPU Scaling: Supports single-GPU and multi-GPU distributed training via the CV_DET_NGPU environment variable. - Use Case: Ask the agent to test mask_rcnn training performance on 8 NVIDIA GPUs; it starts the container, runs both fp16 and fp32 passes, and returns per-iteration timing results. ## Quick Start Ask the agent to test faster_rcnn detection training performance on NVIDIA GPUs with the COCO dataset and pretrained weights mounted.

Frequently Asked Questions about nvidia-cv-detection

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I benchmark object detection training performance on NVIDIA GPUs?

Run the batch_detection.sh script inside the provided NVIDIA CV Docker container with COCO data and pretrained weights mounted. It trains the selected mmdetection models in fp16 and fp32 and writes AVG_ITER_TIME metrics to eval_result.json.

Which detection models does this mmdetection benchmark support?

It supports faster_rcnn, mask_rcnn, cascade_rcnn, retinanet, yolov3, fcos, ssd300, centernet, solo, and swin_mask_rcnn. Each model maps to a predefined mmdetection config and backbone checkpoint selected automatically by the script.

How do I run multi-GPU detection training benchmarks?

Set the CV_DET_NGPU environment variable to the desired GPU count, for example export CV_DET_NGPU=8. The script passes it to torch.distributed.launch as nproc_per_node and tags log directories with the GPU count.

Can I run only fp16 or only fp32 precision in the benchmark?

No, the precision set is fixed to fp16 and fp32 and cannot be changed through user input or environment variables. Every selected model is trained once in each precision.

Why is there no AVG_ITER_TIME in my training log?

AVG_ITER_TIME requires the CustomIterTimerHook to be enabled in the project configuration. Also ensure you read logs created after the current run by using the CV_DET_RUN_MARKER file instead of picking the latest historical log.

What should I do when the mmdetection source directory is not found?

Verify a directory containing configs/ and tools/train.py exists in the image or mounted volumes. If it lives at a custom path, set CV_DET_MMDET_DIR to that location before running the script.