removed gradio

This commit is contained in:
2025-12-23 18:24:40 -07:00
parent 43a34aaf00
commit 8804b45067
12 changed files with 1363 additions and 2271 deletions

View File

@ -14,9 +14,9 @@ RT-DETR (Real-Time Detection Transformer) is Apache 2.0 licensed - **free for co
### 1. Annotate Images
Use the annotation GUI:
Use the Tkinter annotation GUI:
```bash
.venv/bin/python annotation_gui.py
.venv/bin/python tk_annotation_gui.py --images-dir IMAGE/
```
- Load your images from Settings
@ -25,24 +25,14 @@ Use the annotation GUI:
### 2. Train Model
From the GUI:
1. Go to **Training** tab
2. Click "Prepare Dataset" (creates train/valid/test splits)
3. Select **RT-DETR** framework
4. Choose model size:
- `nano` (r18): Fastest, 30-40 FPS on OAK
- `small` (r34): Balanced
- `medium` (r50): More accurate
- `base` (l): Best accuracy, slower
5. Click "Start Training"
Or from command line:
Train from the command line:
```bash
.venv/bin/python train_rtdetr.py \
--dataset-dir dataset_prepared \
--model rtdetr-r18 \
--epochs 100 \
--batch-size 8
.venv/bin/python train_model.py \
--framework rtdetr \
--dataset dataset_prepared \
--output runs/rtdetr_training \
--model-size small \
--epochs 100
```
### 3. Test Model