Open
Add advanced computer vision module for robotics AGI#59
Conversation
- vision/: Core module with 14 sub-packages
- detection/: YOLOv8, DETR, person/hand/grasp detectors + base class
- segmentation/: SemanticSegmentor (DeepLabV3+), InstanceSegmentor (Mask R-CNN),
SAMSegmentor, PanopticSegmentor
- depth/: MiDaS, DepthAnything, stereo SGBM, point cloud generator
- tracking/: SORT/Kalman, ByteTrack, DeepSORT, ReID model
- pose/: Human pose (MediaPipe), 6-DoF object pose, hand pose, 3D lifting
- features/: ORB/SIFT/deep extractors, FLANN/BF matcher, visual odometry
- scene/: Scene analyser, affordance detector, scene graph, spatial relations
- vlm/: CLIP, BLIP-2 VQA, BLIP captioner, OWL-ViT visual grounding
- motion/: Farneback/LK optical flow, MOG2 motion segmentation, predictor
- preprocessing/: ImageEnhancer, Denoiser, SuperResolution (bicubic/ESRGAN)
- utils/: Visualisation, transforms, evaluation metrics, camera utilities
- models/: ModelLoader with registry, SimpleCNN/UNet/DetectionHead
- config/: YAML configs for detection, segmentation, depth
- setup.py: Installable package definition
- requirements.txt: All Python dependencies
- tests/test_vision/: 74-test pytest suite with mocked heavy models
- test_detection.py, test_segmentation.py, test_depth.py,
test_tracking.py, test_pipeline.py
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Co-authored-by: Stacey77 <54900383+Stacey77@users.noreply.github.com>
Copilot
AI
changed the title
[WIP] Add advanced computer vision functions for robotics AGI
Add advanced computer vision module for robotics AGI
Feb 20, 2026
Owner
|
@copilot choose next step |
Contributor
Author
The vision module core is done. The remaining deliverables from the problem statement are ROS2 integration, example notebooks, and benchmarks. Implementing all three now — starting with the ROS2 vision nodes (the most critical for robotics), then notebooks and benchmark scripts. Warning Firewall rules blocked me from connecting to one or more addresses (expand for details)I tried to connect to the following addresses, but was blocked by firewall rules:
If you need me to access, download, or install something from one of these locations, you can either:
|
- ros2_interface/: self-contained ROS2 package with VisionNode, ImageConverter, PointCloudPublisher, service types, launch file, config YAML, and README; all ROS2/cv_bridge imports guarded with try/except for non-ROS environments - notebooks/: seven standalone Python demo scripts (detection, segmentation, depth, tracking, pose, VLM, full pipeline) plus README; work with numpy only - benchmarks/: FPS benchmark (--device/--iterations/--modules CLI) and accuracy benchmark (mAP, MOTA, PCK on synthetic data) with table output and README - tests/test_ros2/: 26 new tests for VisionNode and ImageConverter using mocked rclpy/cv_bridge; all 103 tests pass Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Co-authored-by: Stacey77 <54900383+Stacey77@users.noreply.github.com>
Stacey77
marked this pull request as ready for review
April 4, 2026 00:58
Owner
|
DEPLOY |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Adds a comprehensive
vision/Python package providing modular, lazily-loaded computer vision capabilities (detection, segmentation, depth, tracking, pose, VLM, motion, scene understanding) for a robotics AGI system, along with ROS2 integration, example demo scripts, and benchmark tooling.Module structure
detection/—YOLODetector(ultralytics),DETRDetector(HF transformers),PersonDetector,HandDetector(MediaPipe),GraspDetector; sharedBaseDetectorwith NMSsegmentation/—SemanticSegmentor(DeepLabV3+),InstanceSegmentor(Mask R-CNN),SAMSegmentor,PanopticSegmentordepth/—MonocularDepthEstimator(MiDaS),DepthAnythingEstimator,StereoDepthEstimator,PointCloudGeneratortracking/—MultiObjectTracker(Kalman filter),ByteTracker,DeepSORTTracker,ReIDModelpose/—HumanPoseEstimator(17 keypoints, fall detection),ObjectPoseEstimator(6-DoF PnP),HandPoseEstimator(gestures),Pose3DEstimatorfeatures/—FeatureExtractor(ORB/SIFT/AKAZE/deep),FeatureMatcher,VisualOdometryscene/—SceneAnalyzer,AffordanceDetector,SceneGraph,SpatialRelationsvlm/—CLIPInterface,VisualQA(BLIP-2),ImageCaptioner(BLIP),VisualGrounding(OWL-ViT)motion/—OpticalFlow(Farneback/LK),MotionSegmentor,MotionPredictorpreprocessing/—ImageEnhancer,Denoiser,SuperResolutionutils/— visualization helpers, transforms, metrics (mAP/MOTA/PCK), camera projection utilitiesmodels/—ModelLoaderwith registry + disk cache,SimpleCNN,UNet,DetectionHeadUnified pipeline
All modules are accessible via a single lazy-loading entry point:
ROS2 Integration (
ros2_interface/)VisionNode— ROS2 node subscribing to/camera/image_raw; publishes to/vision/detections,/vision/segmentation,/vision/depth,/vision/poses,/vision/scene_description,/vision/pointcloud/vision/find_object(text query → bbox + confidence) and/vision/analyze_scene(full scene JSON)ImageConverter— ROS Image ↔ numpy conversion with manual byte-level fallback whencv_bridgeis absentPointCloudPublisher— Nx3 numpy array →sensor_msgs/PointCloud2with explicit little-endian encodingvision_node_params.yaml),package.xml, and README includedtry/exceptso the package is importable without a ROS2 installationExample notebooks (
notebooks/)Seven standalone Python demo scripts covering every major module — detection, segmentation, depth, tracking, pose, vision-language, and the full unified pipeline. Scripts run without GPU using numpy-only synthetic images; missing ML dependencies are skipped gracefully.
Benchmarks (
benchmarks/)benchmark_fps.py— measures FPS for each vision module with--device,--iterations, and--modulesCLI args; outputs a results tablebenchmark_accuracy.py— computes mAP, MOTA, and PCK metrics on synthetic dataSupporting files
vision/config/— YAML configs for detection, segmentation, and depth defaultsrequirements.txt/setup.py— full dependency manifesttests/test_vision/— 74 unit tests; heavy model paths mocked so the suite runs without GPUtests/test_ros2/— 26 tests forVisionNodeandImageConverterwith mocked rclpy/cv_bridge; 103 tests total.gitignore— excludes__pycache__, model weights, build artifactsOriginal prompt
Add Advanced Computer Vision Functions for Robotics AGI
Objective
Enhance the Agentic AGI robotics system with comprehensive computer vision capabilities using PyTorch, OpenCV, and state-of-the-art vision models. This should provide robust visual perception for robots to understand and interact with their environment.
Technology Stack
Core Computer Vision Functions to Implement
1. Object Detection (
vision/detection/)Multi-Model Detection System
Features:
Specific Detectors
2. Image Segmentation (
vision/segmentation/)Semantic Segmentation
Instance Segmentation
Interactive Segmentation
3. Depth Estimation (
vision/depth/)Features:
4. Object Tracking (
vision/tracking/)Capabilities:
5. Pose Estimation (
vision/pose/)Human Pose Estimation
Object Pose Estimation
6. Visual Features (
vision/features/)7. Scene Understanding (
vision/scene/)Advanced Features:
8. Vision-Language Models (
vision/vlm/)Capabilities:
9. Optical Flow & Motion (
vision/motion/)