File size: 1,864 Bytes
8bf4052
ec0daf5
 
 
 
8bf4052
ec0daf5
8bf4052
 
 
ec0daf5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
---
title: Inference Studio | AI Vision Explorer
emoji: πŸš€
colorFrom: blue
colorTo: purple
sdk: docker
app_port: 7860
pinned: false
---

# πŸš€ Inference Studio | AI Vision Explorer

A premium, web-based interface for deploying and testing YOLO vision models with advanced controls for ROI (Region of Interest) filtering and confidence range analysis.

![UI Preview](static/ui_preview.png)

## ✨ Features

-   **Interactive ROI Drawing**: Draw detection zones directly on a preview image or video frame.
-   **Confidence Range Filtering**: Test model behavior by specifying both Min and Max confidence thresholds (e.g., visualize only low-confidence detections).
-   **Video Studio**: Full support for video inference with automatic frame extraction for ROI setup and H.264 transcoding for web playback.
-   **Glassmorphism UI**: Modern, dark-themed interface built for a premium developer experience.
-   **Model Management**: Easily swap `.pt` models on the fly.

## πŸ› οΈ Setup

### Prerequisites
-   **Python 3.8+**
-   **FFmpeg**: Required for video processing.
    ```bash
    # Ubuntu/Debian
    sudo apt update && sudo apt install ffmpeg
    ```

### Installation
1.  Clone the repository:
    ```bash
    git clone https://github.com/your-repo/inference-studio.git
    cd inference-studio
    ```
2.  Install dependencies:
    ```bash
    pip install -r requirements.txt
    ```

## πŸš€ Running the Studio

Start the server using the provided shell script:
```bash
chmod +x start.sh
./start.sh
```
The studio will be available at `http://localhost:8000`.

## πŸ“‚ Project Structure

-   `main.py`: FastAPI backend handling inference and video tasks.
-   `static/`: Frontend assets (styles, interactive JS).
-   `templates/`: HTML templates.
-   `uploads/`: Directory structure for models, videos, and results (ignored by Git).

## πŸ“ License
MIT