Skip to content

Easy & Modular Computer Vision Detectors, Trackers & SAM - Run YOLOv9,v8,v7,v6,v5,R,X in under 10 lines of code.

License

Notifications You must be signed in to change notification settings

augmentedstartups/AS-One

Repository files navigation

ASOne

Table of Contents

Asone Library Installation

Install In Docker Container

Prerequisite

  • Make sure you have docker installed in your system. if not reffer to Docker Setup

Installation

  1. Clone the repo
git clone https://github.com/axcelerateai/asone.git
cd asone
  1. Run docker coompose command.
# if you wanna test on gpu system
docker compose run linux-gpu
# if you wanna test on cpu system
docker compose run linux
  1. In docker terminal.
# if using gpu
python main.py [VIDEO_PATH]

# if using cpu
python main.py [VIDEO_PATH] --cpu

Install Locally

  1. Clone the repo
git clone https://github.com/axcelerateai/asone.git
cd asone
  1. Create virtual env.
python3 -m venv .env
  1. Activate venv
# linux
source .env/bin/activate

# Windows

.env\Scripts\activate
  1. Intall pre-requisite
pip install numpy Cython
# for windows
pip install -e git+https://github.com/samson-wang/cython_bbox.git#egg=cython-bbox

# for linux
pip install cython-bbox
  1. Install torch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
  1. Install asone
pip install .
  1. Test it by runiing main.py
# if using gpu
python main.py [VIDEO_PATH]

# if using cpu
python main.py [VIDEO_PATH] --cpu