SAHI | Машинное зрение для обнаружения крупномасштабных объектов и сегментации экземпляров

Демки

SAHI — библиотека машинного зрения для обнаружения крупномасштабных объектов и сегментации экземпляров.

Установка:

pip install sahi

Код

pip install sahi

На Windows Shapely необходимо установить через Conda:

conda install -c conda-forge shapely

Код

conda install -c conda-forge shapely

Установка, необходимой версии py torch и torch vision (cuda 11.3 для detectron2, cuda 11.7 для forrest).:

conda install pytorch=1.10.2 torchvision=0.11.3 cudatoolkit=11.3 -c pytorch
conda install pytorch=1.13.1 torchvision=0.14.1 pytorch-cuda=11.7 -c pytorch -c nvidia

Код

conda install pytorch=1.10.2 torchvision=0.11.3 cudatoolkit=11.3 -c pytorch

conda install pytorch=1.13.1 torchvision=0.14.1 pytorch-cuda=11.7 -c pytorch -c nvidia

Установка фреймворков обнаружения (yolov5, mmdet, detectron2 и huggingface):

pip install yolov5==7.0.4
pip install mmcv-full==1.7.0 -f [URL]https://download.openmmlab.com/mmcv/dist/cu117/torch1.13.0/index.html[/URL]
pip install mmdet==2.26.0
pip install detectron2 -f [URL]https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html[/URL]
pip install transformers timm

Код

pip install yolov5==7.0.4

pip install mmcv-full==1.7.0 -f [URL]https://download.openmmlab.com/mmcv/dist/cu117/torch1.13.0/index.html[/URL]

pip install mmdet==2.26.0

pip install detectron2 -f [URL]https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html[/URL]

pip install transformers timm

Ссылка на GitHub с подробными инструкциями для работы с SAHI: https://github.com/obss/sahi