20 lines (20 loc) · 999 Bytes

Инструкция по запуску инференса

Создать папку ckpt.
```
mkdir ckpt
```
Скачать веса моделей https://vicserver.crs4.it/atlantanet/, переместить их в папку ckpt.
Создать виртуальное окружение
```
python -m venv venv
```
Активировать виртуальное окружение
```
.\venv\Scripts\Activate.ps1
```
Установить необходимые зависимости
```
pip install -r .\requirements.txt
```

Запустить инференс

python inference_atlanta_net.py --pth ckpt/resnet101_atlantalayout.pth --img data/atlantalayout/test/img/2t7WUuJeko7_c2e11b94c07a4d6c85cc60286f586a02_equi.png

Предобработка панорамного изображения

python inference_atlanta_net.py --pth ckpt/resnet101_atlantalayout.pth --img .\preprocesspanoram.jpg