Загрузка...

SAHI | Computer vision for large-scale object detection and instance segmentation

Thread in Python created by APT29388 Dec 8, 2024. 151 view

  1. APT29388
    APT29388 Topic starter Dec 8, 2024 ГУРУ ИНВАЙТА - lolz.live/threads/8567181 :admin:
    [IMG]
    [IMG]
    [IMG]

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

    Установка:
    Code
    pip install sahi
    На Windows Shapely необходимо установить через Conda:
    Code
    conda install -c conda-forge shapely
    Установка, необходимой версии py torch и torch vision (cuda 11.3 для detectron2, cuda 11.7 для forrest).:
    Code
    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):
    Code
    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
     
Top
Loading...