AOSSIE-Org / AOSSIE-Org/PictoPy

Feat:AUTO Mode for YOLO Model Selection

未關閉
#903 4 則留言 0 個 reaction 已指派 0 人 在 GitHub 檢視
主要語言
Python
星號
283
分支
679
平均合併
7 天 2 小時
30 天內合併 PR
3

描述

### Describe the feature

Currently, PictoPy requires users to manually select the YOLO model size (nano, small, or medium). While this offers flexibility, it requires users to understand performance–accuracy trade-offs and system limitations.

**Introduce an AUTO mode that dynamically selects the most suitable YOLO model at runtime based on contextual information such as input characteristics and system resources.**

### Dynamic Decision Factors

**When AUTO mode is enabled, the system should evaluate multiple signals before choosing the model:**

- Image resolution (small images → lighter models)
- Estimated number of faces or objects
- User priority (speed vs accuracy)
- GPU availability and free memory
- Task type (face detection or object detection)

**This allows the system to adapt model choice per request instead of relying on a static configuration.**

### Multi-Stage Selection Strategy

**AUTO mode can follow a staged decision approach:**

- Pre-inference checks
Use lightweight models for small images or limited hardware resources.

- Fast probe detection
Run a quick detection pass using a lightweight model to estimate face/object count.

- Runtime fallback
Automatically downgrade the model if inference fails (e.g., GPU out-of-memory).

### User Benefits

- Smarter default behavior without manual tuning
- Improved performance and resource utilization
- Reduced inference failures on constrained systems
- Better overall user experience, especially for new users

### Backward Compatibility

- Existing manual model selection (nano, small, medium) remains unchanged
- AUTO mode is optional and opt-in
- No breaking changes to current workflows , raise an issue for gsoc style in github

### Add ScreenShots

Image

### Record

- [x] I agree to follow this project's Code of Conduct
- [x] I want to work on this issue

貢獻指南

開啟貢獻指南

評估

這個 Issue 還沒有評估資料。

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。