senior-computer-vision

Trains computer vision models for detection and segmentation with PyTorch and YOLO/CNN architectures.

Updated Mar 12, 2026
One-click install
npx skills add https://github.com/Fantasia1999/claude-skills-zh --skill senior-computer-vision-fantasia1999
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: senior-computer-vision
Source: https://github.com/Fantasia1999/claude-skills-zh/tree/main/translations/engineering-team/senior-computer-vision
Command: npx skills add https://github.com/Fantasia1999/claude-skills-zh --skill senior-computer-vision-fantasia1999

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires torchvision, ultralytics, detectron2, mmdetection, onnx, tensorrt, opencv-python, albumentations, pycocotools, and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill streamlines the development and deployment of production-grade computer vision systems for tasks like object detection and image segmentation.

Core Features & Use Cases

  • End-to-End Workflows: Guides you through dataset preparation, model training, optimization, and deployment.
  • Architecture Selection: Provides recommendations for various vision tasks and deployment targets.
  • Use Case: Deploy a real-time object detection system on an edge device by optimizing a YOLO model for TensorRT.

Quick Start

Use the senior-computer-vision skill to prepare a custom dataset for object detection training.

Frequently Asked Questions about senior-computer-vision

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

FAQPage Schema
How do I deploy a YOLO object detection model using TensorRT?

To deploy a YOLO object detection model using TensorRT, you export the trained model to ONNX format and use TensorRT to optimize inference for edge devices. This skill provides end-to-end workflows for this exact optimization process.

What is the best way to train a custom image segmentation model?

The best way to train a custom image segmentation model is using frameworks like Detectron2 or MMDetection with architectures like Mask R-CNN or SAM. This skill guides dataset preparation, model training, and architecture selection for segmentation tasks.

Can I use Detectron2 with PyTorch for object detection pipelines?

Yes, you can use Detectron2 with PyTorch to build object detection pipelines. This skill facilitates building detection pipelines and training custom models using frameworks like PyTorch, Ultralytics, and Detectron2 for Vision AI systems.

How do I prepare a custom dataset for object detection training?

You prepare a custom dataset for object detection training by formatting annotations for frameworks like Ultralytics or Detectron2. This skill streamlines dataset preparation, model training, optimization, and deployment workflows for production-grade vision systems.

Does ONNX support Vision Transformer architectures for inference optimization?

Yes, ONNX supports Vision Transformer architectures for inference optimization. This skill covers CNN and Vision Transformer architectures, facilitating model optimization and deployment using ONNX and TensorRT for visual AI systems.

Why use TensorRT for computer vision model deployment over native PyTorch?

You use TensorRT for computer vision model deployment over native PyTorch to achieve optimized inference on edge devices. This skill guides you through exporting models to ONNX and deploying them with TensorRT for production-grade vision systems.