Files
annotations/Azaion.AI
Alex Bezdieniezhnykh 62623b7123 add ramdisk, load AI model to ramdisk and start recognition from it
rewrite zmq to DEALER and ROUTER
add GET_USER command to get CurrentUser from Python
all auth is on the python side
inference run and validate annotations on python
2025-01-29 17:45:26 +02:00
..

Azaion AI

Azaion AI is a worker written on cython (c-compilable python) which listens to socket and rabbit queue. It accepts commands om a format:

  • CommandType: Inference / Load
  • Filename

And correspondingly do inference or just load encrypted file from the API. Results (file or annotations) is putted to the other queue, or the same socket, depending on the command source.

Installation

Install libs

https://www.python.org/downloads/

Windows

Linux

    sudo apt install nvidia-driver-535
    
    wget https://developer.download.nvidia.com/compute/cudnn/9.2.0/local_installers/cudnn-local-repo-ubuntu2204-9.2.0_1.0-1_amd64.deb
    sudo dpkg -i cudnn-local-repo-ubuntu2204-9.2.0_1.0-1_amd64.deb
    
    sudo cp /var/cudnn-local-repo-ubuntu2204-9.2.0/cudnn-*-keyring.gpg /usr/share/keyrings/
    sudo apt-get update
    sudo apt-get -y install cudnn nvidia-cuda-toolkit -y
    nvcc --version

Install dependencies

1. Install python with max version 3.11. Pytorch for now supports 3.11 max

Make sure that your virtual env is installed with links to the global python packages and headers, like this:

    python -m venv --system-site-packages venv

This is crucial for the build because build needs Python.h header and other files.

    python -m pip install --upgrade pip
    pip install --upgrade huggingface_hub
    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
    pip install ultralytics
    
    pip uninstall -y opencv-python
    pip install opencv-python cython msgpack cryptography rstream pika zmq pyjwt

In case of fbgemm.dll error (Windows specific):

  • copypaste libomp140.x86_64.dll to C:\Windows\System32

Build

``` python setup.py build_ext --inplace ```