SAHI — библиотека машинного зрения для обнаружения крупномасштабных объектов и сегментации экземпляров.
Установка:
$ pip install sahi
On Windows, Shapely needs to be installed via Conda:
$ conda install -c conda-forge shapely
Install your desired version of pytorch and torchvision (cuda 11.3 for detectron2, cuda 11.7 for rest):
$ 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
Install your desired detection framework (yolov5):
$ pip install yolov5==7.0.4
Install your desired detection framework (mmdet):
$ pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13.0/index.html
$ pip install mmdet==2.26.0
Install your desired detection framework (detectron2):
$ pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
Install your desired detection framework (huggingface):
$ pip install transformers timm
#python #github #soft