removed gradio
This commit is contained in:
@ -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
|
||||
|
||||
Reference in New Issue
Block a user