Add GTA-UAV-LR annotation script + dataset documentation

- Add scripts/run_gta_uav.py for GTA-UAV-LR (48K images, GTA V synthetic)
- 14 segmentation classes: 11 base + bare soil, rooftop, swimming pool
- Fix source filter to recognize "satellite" folder (alongside "DB")
- Document GTA-UAV characteristics in segmentation_class_analysis.md
- Update README and CLAUDE.md with GTA-UAV support

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-17 21:55:56 +03:00
parent 27eb315903
commit 143a837c03
5 changed files with 210 additions and 4 deletions

View File

@@ -10,10 +10,11 @@
| **CHMv2** | DINOv3-ViTL16 (337M, FP32) | grayscale [256x256] | 31.7 img/s |
| **Consolidate** | SafeTensors (CPU) | `.safetensors` per image | ~5000 img/s |
| Датасет | Изображения | Классы сегментации | Скрипт |
| Датасет | Изображения | Классы сегм. | Скрипт |
|:---|:---|:---|:---|
| **World-UAV** | 973K (486K DB + 486K query) | 11 | `python -m src.main` |
| **UAV_VisLoc** | 81K (75K DB + 6.7K drone) | 16 | `python scripts/run_uav_visloc.py` |
| **GTA-UAV-LR** | 48K (15K sat + 34K drone) | 14 | `python scripts/run_gta_uav.py` |
## Quick Start
@@ -24,6 +25,9 @@ python -m src.main
# UAV_VisLoc
python scripts/run_uav_visloc.py
# GTA-UAV-LR
python scripts/run_gta_uav.py
# Тесты (143 шт, без GPU)
python -m pytest src/tests/ -v
```
@@ -67,6 +71,7 @@ python -m pytest src/tests/ -v
│ └── tests/ # 143 теста (pytest)
├── scripts/
│ ├── run_uav_visloc.py # Запуск для UAV_VisLoc (16 классов)
│ ├── run_gta_uav.py # Запуск для GTA-UAV-LR (14 классов)
│ └── migrate_layout.py # Миграция со старого prefix-формата
└── docs/
├── segmentation_class_analysis.md # Анализ классов сегментации (11 классов)
@@ -135,6 +140,21 @@ UAV_VisLoc содержит больше сельских/прибрежных
> Threshold снижен до 0.1 (vs 0.15 для World-UAV) — UAV_VisLoc содержит больше тёмных/низкоконтрастных сцен.
### GTA-UAV-LR: 14 классов
Синтетический датасет из GTA V. Особенности:
- Satellite в RGBA (alpha = граница карты) → автоматически конвертируется в RGB
- Drone 512x384 (не квадратные) → resize в 512x512
- 6 высот полёта: 100, 200, 300, 400, 500, 600 м
Классы = 11 базовых + 3 специфических для GTA:
| ID | Промпт | Зачем |
|:--:|:---|:---|
| 11 | `bare soil and plowed field` | Пустыри, стройплощадки (много в пригородах GTA) |
| 12 | `roof and rooftop` | Плоские крыши, вертолётные площадки |
| 13 | `swimming pool` | Бассейны — характерный landmark пригородов GTA |
Подробный анализ выбора классов: [`docs/segmentation_class_analysis.md`](docs/segmentation_class_analysis.md)
### hardware.gin