flashstudio 0.1.0__tar.gz → 0.1.2__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (27) hide show
  1. {flashstudio-0.1.0 → flashstudio-0.1.2}/PKG-INFO +43 -17
  2. {flashstudio-0.1.0 → flashstudio-0.1.2}/README.md +41 -16
  3. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/launcher.py +6 -1
  4. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio.egg-info/PKG-INFO +43 -17
  5. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio.egg-info/requires.txt +1 -0
  6. {flashstudio-0.1.0 → flashstudio-0.1.2}/pyproject.toml +2 -1
  7. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/__init__.py +0 -0
  8. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/app.py +0 -0
  9. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/cli.py +0 -0
  10. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/components/__init__.py +0 -0
  11. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/components/sidebar.py +0 -0
  12. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/components/styles.py +0 -0
  13. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/components/wizard.py +0 -0
  14. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/pages/__init__.py +0 -0
  15. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/pages/dashboard.py +0 -0
  16. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/pages/data.py +0 -0
  17. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/pages/export.py +0 -0
  18. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/pages/inference.py +0 -0
  19. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/pages/model.py +0 -0
  20. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/pages/training.py +0 -0
  21. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/utils/__init__.py +0 -0
  22. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio/utils/device.py +0 -0
  23. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio.egg-info/SOURCES.txt +0 -0
  24. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio.egg-info/dependency_links.txt +0 -0
  25. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio.egg-info/entry_points.txt +0 -0
  26. {flashstudio-0.1.0 → flashstudio-0.1.2}/flashstudio.egg-info/top_level.txt +0 -0
  27. {flashstudio-0.1.0 → flashstudio-0.1.2}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flashstudio
3
- Version: 0.1.0
3
+ Version: 0.1.2
4
4
  Summary: Interactive Training & Inference UI for FlashDet — runs on Google Colab
5
5
  Author: Gaurav14cs17
6
6
  License: Apache-2.0
@@ -19,6 +19,7 @@ Requires-Dist: plotly>=5.0
19
19
  Requires-Dist: pandas>=1.5
20
20
  Requires-Dist: pillow>=9.0
21
21
  Requires-Dist: numpy>=1.21
22
+ Requires-Dist: pyngrok>=6.0
22
23
  Provides-Extra: full
23
24
  Requires-Dist: pyngrok>=6.0; extra == "full"
24
25
  Requires-Dist: torch>=2.0; extra == "full"
@@ -31,6 +32,10 @@ Requires-Dist: pytest; extra == "dev"
31
32
 
32
33
  # ⚡ FlashStudio
33
34
 
35
+ [![PyPI version](https://badge.fury.io/py/flashstudio.svg)](https://pypi.org/project/flashstudio/)
36
+ [![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/)
37
+ [![License: Apache-2.0](https://img.shields.io/badge/License-Apache%202.0-green.svg)](https://opensource.org/licenses/Apache-2.0)
38
+
34
39
  **Interactive Training & Inference UI for FlashDet** — runs locally or on Google Colab with a Streamlit interface.
35
40
 
36
41
  <p align="center">
@@ -39,22 +44,33 @@ Requires-Dist: pytest; extra == "dev"
39
44
 
40
45
  ## Features
41
46
 
42
- - 🏋️ **Training Dashboard** — Real-time monitoring with live loss curves, visualizations, GT verification
47
+ - 🏋️ **Training Dashboard** — Real-time monitoring with live loss curves, per-epoch visualizations, GT verification
43
48
  - 🧠 **Model Config** — All 6 FlashDet sizes + YOLOv8/v9/v10/v11/YOLOX with accurate params
44
49
  - 🔍 **Inference Pipeline** — 4-step wizard: Model → Data → Zone → Run (17 solutions, 6 trackers)
45
50
  - 📤 **Export** — ONNX export with FP16 auto-generated weights
46
- - 📦 **Data** — Native `flashdet download` datasets + custom upload
51
+ - 📦 **Data** — Native `flashdet download` datasets + custom upload (COCO/VOC/YOLO formats)
52
+ - 📊 **Dashboard** — Overview with recent training runs from workspace
47
53
  - 🚀 **Colab Support** — ngrok tunneling for remote access
48
54
 
49
- ## Quick Start
55
+ ## Install
56
+
57
+ ```bash
58
+ pip install flashstudio
59
+ ```
60
+
61
+ ### Install with all dependencies (PyTorch + OpenCV + ngrok)
62
+
63
+ ```bash
64
+ pip install flashstudio[full]
65
+ ```
50
66
 
51
- ### Install from GitHub
67
+ ### Install FlashDet (required for training/inference)
52
68
 
53
69
  ```bash
54
- pip install git+https://github.com/FlashVision/FlashStudio.git
70
+ pip install git+https://github.com/FlashVision/FlashDet.git
55
71
  ```
56
72
 
57
- ### Install locally (development)
73
+ ### Development install
58
74
 
59
75
  ```bash
60
76
  git clone https://github.com/FlashVision/FlashStudio.git
@@ -62,23 +78,21 @@ cd FlashStudio
62
78
  pip install -e .
63
79
  ```
64
80
 
65
- ### Install with all dependencies (FlashDet + PyTorch)
81
+ ## Quick Start
82
+
83
+ ### CLI
66
84
 
67
85
  ```bash
68
- pip install "flashstudio[full] @ git+https://github.com/FlashVision/FlashStudio.git"
86
+ flashstudio --port 8501
69
87
  ```
70
88
 
71
- ### Run
89
+ ### Python (Streamlit directly)
72
90
 
73
91
  ```bash
74
- # CLI
75
- flashstudio --port 8501
76
-
77
- # Or directly
78
92
  streamlit run flashstudio/app.py
79
93
  ```
80
94
 
81
- ### Python API (for Colab)
95
+ ### Python API (Google Colab)
82
96
 
83
97
  ```python
84
98
  from flashstudio import launch
@@ -92,6 +106,18 @@ launch() # Opens ngrok tunnel in Colab, localhost otherwise
92
106
  | Training | Train FlashDet models | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/FlashVision/FlashStudio/blob/main/notebooks/FlashStudio_Train.ipynb) |
93
107
  | Inference | Run detection on images/video | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/FlashVision/FlashStudio/blob/main/notebooks/FlashStudio_Inference.ipynb) |
94
108
 
109
+ ## Supported Models
110
+
111
+ | Model | Params | Best For |
112
+ |-------|--------|----------|
113
+ | FlashDet-Pico | ~298K | Edge / MCU |
114
+ | FlashDet-Nano | ~790K | Embedded / IoT |
115
+ | FlashDet-Small | ~1.8M | General purpose |
116
+ | FlashDet-Medium | ~3.6M | High accuracy |
117
+ | FlashDet-Large | ~5.8M | High accuracy |
118
+ | FlashDet-X | ~9.0M | Max accuracy / Server |
119
+ | YOLOv8/v9/v10/v11/YOLOX | Varies | General YOLO |
120
+
95
121
  ## Architecture
96
122
 
97
123
  ```
@@ -102,7 +128,7 @@ FlashStudio/
102
128
  │ ├── launcher.py # Colab/local launcher with ngrok
103
129
  │ ├── cli.py # CLI entrypoint
104
130
  │ ├── pages/
105
- │ │ ├── dashboard.py # Overview + recent runs
131
+ │ │ ├── dashboard.py # Overview + recent training runs
106
132
  │ │ ├── data.py # Dataset upload/download
107
133
  │ │ ├── model.py # Architecture & hyperparameter config
108
134
  │ │ ├── training.py # Training monitor (reads real workspace)
@@ -125,7 +151,7 @@ FlashStudio/
125
151
  ## Requirements
126
152
 
127
153
  - Python >= 3.9
128
- - FlashDet (install separately: `pip install git+https://github.com/FlashVision/FlashDet.git`)
154
+ - FlashDet (`pip install git+https://github.com/FlashVision/FlashDet.git`)
129
155
  - GPU recommended for training (T4 or better)
130
156
 
131
157
  ## License
@@ -1,5 +1,9 @@
1
1
  # ⚡ FlashStudio
2
2
 
3
+ [![PyPI version](https://badge.fury.io/py/flashstudio.svg)](https://pypi.org/project/flashstudio/)
4
+ [![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/)
5
+ [![License: Apache-2.0](https://img.shields.io/badge/License-Apache%202.0-green.svg)](https://opensource.org/licenses/Apache-2.0)
6
+
3
7
  **Interactive Training & Inference UI for FlashDet** — runs locally or on Google Colab with a Streamlit interface.
4
8
 
5
9
  <p align="center">
@@ -8,22 +12,33 @@
8
12
 
9
13
  ## Features
10
14
 
11
- - 🏋️ **Training Dashboard** — Real-time monitoring with live loss curves, visualizations, GT verification
15
+ - 🏋️ **Training Dashboard** — Real-time monitoring with live loss curves, per-epoch visualizations, GT verification
12
16
  - 🧠 **Model Config** — All 6 FlashDet sizes + YOLOv8/v9/v10/v11/YOLOX with accurate params
13
17
  - 🔍 **Inference Pipeline** — 4-step wizard: Model → Data → Zone → Run (17 solutions, 6 trackers)
14
18
  - 📤 **Export** — ONNX export with FP16 auto-generated weights
15
- - 📦 **Data** — Native `flashdet download` datasets + custom upload
19
+ - 📦 **Data** — Native `flashdet download` datasets + custom upload (COCO/VOC/YOLO formats)
20
+ - 📊 **Dashboard** — Overview with recent training runs from workspace
16
21
  - 🚀 **Colab Support** — ngrok tunneling for remote access
17
22
 
18
- ## Quick Start
23
+ ## Install
24
+
25
+ ```bash
26
+ pip install flashstudio
27
+ ```
28
+
29
+ ### Install with all dependencies (PyTorch + OpenCV + ngrok)
30
+
31
+ ```bash
32
+ pip install flashstudio[full]
33
+ ```
19
34
 
20
- ### Install from GitHub
35
+ ### Install FlashDet (required for training/inference)
21
36
 
22
37
  ```bash
23
- pip install git+https://github.com/FlashVision/FlashStudio.git
38
+ pip install git+https://github.com/FlashVision/FlashDet.git
24
39
  ```
25
40
 
26
- ### Install locally (development)
41
+ ### Development install
27
42
 
28
43
  ```bash
29
44
  git clone https://github.com/FlashVision/FlashStudio.git
@@ -31,23 +46,21 @@ cd FlashStudio
31
46
  pip install -e .
32
47
  ```
33
48
 
34
- ### Install with all dependencies (FlashDet + PyTorch)
49
+ ## Quick Start
50
+
51
+ ### CLI
35
52
 
36
53
  ```bash
37
- pip install "flashstudio[full] @ git+https://github.com/FlashVision/FlashStudio.git"
54
+ flashstudio --port 8501
38
55
  ```
39
56
 
40
- ### Run
57
+ ### Python (Streamlit directly)
41
58
 
42
59
  ```bash
43
- # CLI
44
- flashstudio --port 8501
45
-
46
- # Or directly
47
60
  streamlit run flashstudio/app.py
48
61
  ```
49
62
 
50
- ### Python API (for Colab)
63
+ ### Python API (Google Colab)
51
64
 
52
65
  ```python
53
66
  from flashstudio import launch
@@ -61,6 +74,18 @@ launch() # Opens ngrok tunnel in Colab, localhost otherwise
61
74
  | Training | Train FlashDet models | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/FlashVision/FlashStudio/blob/main/notebooks/FlashStudio_Train.ipynb) |
62
75
  | Inference | Run detection on images/video | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/FlashVision/FlashStudio/blob/main/notebooks/FlashStudio_Inference.ipynb) |
63
76
 
77
+ ## Supported Models
78
+
79
+ | Model | Params | Best For |
80
+ |-------|--------|----------|
81
+ | FlashDet-Pico | ~298K | Edge / MCU |
82
+ | FlashDet-Nano | ~790K | Embedded / IoT |
83
+ | FlashDet-Small | ~1.8M | General purpose |
84
+ | FlashDet-Medium | ~3.6M | High accuracy |
85
+ | FlashDet-Large | ~5.8M | High accuracy |
86
+ | FlashDet-X | ~9.0M | Max accuracy / Server |
87
+ | YOLOv8/v9/v10/v11/YOLOX | Varies | General YOLO |
88
+
64
89
  ## Architecture
65
90
 
66
91
  ```
@@ -71,7 +96,7 @@ FlashStudio/
71
96
  │ ├── launcher.py # Colab/local launcher with ngrok
72
97
  │ ├── cli.py # CLI entrypoint
73
98
  │ ├── pages/
74
- │ │ ├── dashboard.py # Overview + recent runs
99
+ │ │ ├── dashboard.py # Overview + recent training runs
75
100
  │ │ ├── data.py # Dataset upload/download
76
101
  │ │ ├── model.py # Architecture & hyperparameter config
77
102
  │ │ ├── training.py # Training monitor (reads real workspace)
@@ -94,7 +119,7 @@ FlashStudio/
94
119
  ## Requirements
95
120
 
96
121
  - Python >= 3.9
97
- - FlashDet (install separately: `pip install git+https://github.com/FlashVision/FlashDet.git`)
122
+ - FlashDet (`pip install git+https://github.com/FlashVision/FlashDet.git`)
98
123
  - GPU recommended for training (T4 or better)
99
124
 
100
125
  ## License
@@ -36,7 +36,12 @@ def launch(port: int = 8501, share: bool = True, ngrok_token: str | None = None)
36
36
 
37
37
  def _launch_colab(app_path: str, port: int, share: bool, ngrok_token: str | None):
38
38
  """Launch in Google Colab with ngrok tunnel."""
39
- from pyngrok import ngrok, conf
39
+ try:
40
+ from pyngrok import ngrok, conf
41
+ except ImportError:
42
+ print("Installing pyngrok...")
43
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", "pyngrok"])
44
+ from pyngrok import ngrok, conf
40
45
 
41
46
  token = ngrok_token or os.environ.get("NGROK_TOKEN", "")
42
47
  if token:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flashstudio
3
- Version: 0.1.0
3
+ Version: 0.1.2
4
4
  Summary: Interactive Training & Inference UI for FlashDet — runs on Google Colab
5
5
  Author: Gaurav14cs17
6
6
  License: Apache-2.0
@@ -19,6 +19,7 @@ Requires-Dist: plotly>=5.0
19
19
  Requires-Dist: pandas>=1.5
20
20
  Requires-Dist: pillow>=9.0
21
21
  Requires-Dist: numpy>=1.21
22
+ Requires-Dist: pyngrok>=6.0
22
23
  Provides-Extra: full
23
24
  Requires-Dist: pyngrok>=6.0; extra == "full"
24
25
  Requires-Dist: torch>=2.0; extra == "full"
@@ -31,6 +32,10 @@ Requires-Dist: pytest; extra == "dev"
31
32
 
32
33
  # ⚡ FlashStudio
33
34
 
35
+ [![PyPI version](https://badge.fury.io/py/flashstudio.svg)](https://pypi.org/project/flashstudio/)
36
+ [![Python 3.9+](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/downloads/)
37
+ [![License: Apache-2.0](https://img.shields.io/badge/License-Apache%202.0-green.svg)](https://opensource.org/licenses/Apache-2.0)
38
+
34
39
  **Interactive Training & Inference UI for FlashDet** — runs locally or on Google Colab with a Streamlit interface.
35
40
 
36
41
  <p align="center">
@@ -39,22 +44,33 @@ Requires-Dist: pytest; extra == "dev"
39
44
 
40
45
  ## Features
41
46
 
42
- - 🏋️ **Training Dashboard** — Real-time monitoring with live loss curves, visualizations, GT verification
47
+ - 🏋️ **Training Dashboard** — Real-time monitoring with live loss curves, per-epoch visualizations, GT verification
43
48
  - 🧠 **Model Config** — All 6 FlashDet sizes + YOLOv8/v9/v10/v11/YOLOX with accurate params
44
49
  - 🔍 **Inference Pipeline** — 4-step wizard: Model → Data → Zone → Run (17 solutions, 6 trackers)
45
50
  - 📤 **Export** — ONNX export with FP16 auto-generated weights
46
- - 📦 **Data** — Native `flashdet download` datasets + custom upload
51
+ - 📦 **Data** — Native `flashdet download` datasets + custom upload (COCO/VOC/YOLO formats)
52
+ - 📊 **Dashboard** — Overview with recent training runs from workspace
47
53
  - 🚀 **Colab Support** — ngrok tunneling for remote access
48
54
 
49
- ## Quick Start
55
+ ## Install
56
+
57
+ ```bash
58
+ pip install flashstudio
59
+ ```
60
+
61
+ ### Install with all dependencies (PyTorch + OpenCV + ngrok)
62
+
63
+ ```bash
64
+ pip install flashstudio[full]
65
+ ```
50
66
 
51
- ### Install from GitHub
67
+ ### Install FlashDet (required for training/inference)
52
68
 
53
69
  ```bash
54
- pip install git+https://github.com/FlashVision/FlashStudio.git
70
+ pip install git+https://github.com/FlashVision/FlashDet.git
55
71
  ```
56
72
 
57
- ### Install locally (development)
73
+ ### Development install
58
74
 
59
75
  ```bash
60
76
  git clone https://github.com/FlashVision/FlashStudio.git
@@ -62,23 +78,21 @@ cd FlashStudio
62
78
  pip install -e .
63
79
  ```
64
80
 
65
- ### Install with all dependencies (FlashDet + PyTorch)
81
+ ## Quick Start
82
+
83
+ ### CLI
66
84
 
67
85
  ```bash
68
- pip install "flashstudio[full] @ git+https://github.com/FlashVision/FlashStudio.git"
86
+ flashstudio --port 8501
69
87
  ```
70
88
 
71
- ### Run
89
+ ### Python (Streamlit directly)
72
90
 
73
91
  ```bash
74
- # CLI
75
- flashstudio --port 8501
76
-
77
- # Or directly
78
92
  streamlit run flashstudio/app.py
79
93
  ```
80
94
 
81
- ### Python API (for Colab)
95
+ ### Python API (Google Colab)
82
96
 
83
97
  ```python
84
98
  from flashstudio import launch
@@ -92,6 +106,18 @@ launch() # Opens ngrok tunnel in Colab, localhost otherwise
92
106
  | Training | Train FlashDet models | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/FlashVision/FlashStudio/blob/main/notebooks/FlashStudio_Train.ipynb) |
93
107
  | Inference | Run detection on images/video | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/FlashVision/FlashStudio/blob/main/notebooks/FlashStudio_Inference.ipynb) |
94
108
 
109
+ ## Supported Models
110
+
111
+ | Model | Params | Best For |
112
+ |-------|--------|----------|
113
+ | FlashDet-Pico | ~298K | Edge / MCU |
114
+ | FlashDet-Nano | ~790K | Embedded / IoT |
115
+ | FlashDet-Small | ~1.8M | General purpose |
116
+ | FlashDet-Medium | ~3.6M | High accuracy |
117
+ | FlashDet-Large | ~5.8M | High accuracy |
118
+ | FlashDet-X | ~9.0M | Max accuracy / Server |
119
+ | YOLOv8/v9/v10/v11/YOLOX | Varies | General YOLO |
120
+
95
121
  ## Architecture
96
122
 
97
123
  ```
@@ -102,7 +128,7 @@ FlashStudio/
102
128
  │ ├── launcher.py # Colab/local launcher with ngrok
103
129
  │ ├── cli.py # CLI entrypoint
104
130
  │ ├── pages/
105
- │ │ ├── dashboard.py # Overview + recent runs
131
+ │ │ ├── dashboard.py # Overview + recent training runs
106
132
  │ │ ├── data.py # Dataset upload/download
107
133
  │ │ ├── model.py # Architecture & hyperparameter config
108
134
  │ │ ├── training.py # Training monitor (reads real workspace)
@@ -125,7 +151,7 @@ FlashStudio/
125
151
  ## Requirements
126
152
 
127
153
  - Python >= 3.9
128
- - FlashDet (install separately: `pip install git+https://github.com/FlashVision/FlashDet.git`)
154
+ - FlashDet (`pip install git+https://github.com/FlashVision/FlashDet.git`)
129
155
  - GPU recommended for training (T4 or better)
130
156
 
131
157
  ## License
@@ -3,6 +3,7 @@ plotly>=5.0
3
3
  pandas>=1.5
4
4
  pillow>=9.0
5
5
  numpy>=1.21
6
+ pyngrok>=6.0
6
7
 
7
8
  [colab]
8
9
  pyngrok>=6.0
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "flashstudio"
7
- version = "0.1.0"
7
+ version = "0.1.2"
8
8
  description = "Interactive Training & Inference UI for FlashDet — runs on Google Colab"
9
9
  readme = "README.md"
10
10
  license = {text = "Apache-2.0"}
@@ -25,6 +25,7 @@ dependencies = [
25
25
  "pandas>=1.5",
26
26
  "pillow>=9.0",
27
27
  "numpy>=1.21",
28
+ "pyngrok>=6.0",
28
29
  ]
29
30
 
30
31
  [project.optional-dependencies]
File without changes