Skip to content

feat: Add YOLOE-26-prompt-based#217

Merged
jamjamjon merged 3 commits intomainfrom
eee
Jan 25, 2026
Merged

feat: Add YOLOE-26-prompt-based#217
jamjamjon merged 3 commits intomainfrom
eee

Conversation

@jamjamjon
Copy link
Copy Markdown
Owner

@jamjamjon jamjamjon commented Jan 25, 2026

Major Features:

  • Add SAM3-Image v2 (3 ONNX) with integrated geometry encoder.
  • Added YOLOE-26 (prompt-based) model and updated YOLOE-26 (prompt-free, one-to-one) variant
  • Added frame-skipping support for webcam input
  • Added support for an infinite progress bar
  • Updated YOLO preprocessing to use Letterbox on both CPU and CUDA, addressing performance issues with YOLOE-26 models

Major Features:
- Add SAM3-Image v2 (3 ONNX) with integrated geometry encoder
- Implement few-images+many-prompts optimized pipeline
- Add frame skipping support for video processing
- Add infinity progress bar support

Performance Optimizations:
- Optimize text_cache to be backend-agnostic (CPU/CUDA/TensorRT)
- Minimize device-to-CPU transfers and type conversions
- Reduce decoder call count from images*prompts to prompts
- Refactor postprocess to use zero-copy views

Code Quality:
- Remove TODO comments and cleanup pipeline code
- Format Python inference scripts with proper line breaks
- Collapse v1 content in README for better organization
- Update documentation links and examples

New Files:
- scripts/sam3-image/export_v2.py: ONNX export for v2 models
- scripts/sam3-image/inference_v2.py: Inference script for v2 models
- src/processor/image/cuda/kernels/: CUDA kernel implementations
- tests/: Test suite additions

Breaking Changes:
- Update default batch sizes and configurations
- Modify progress bar to support infinity display
- Update dataloader to support frame skipping
@jamjamjon jamjamjon changed the title Eee feat: Add YOLOE-26-prompt-based Jan 25, 2026
@jamjamjon jamjamjon merged commit 8b7619a into main Jan 25, 2026
17 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant