Tools

This application is used for exporting Yolo V5, V6 and V7 object detection models for OAKs.

Running the app locally

To run the application locally you need to do 2 additional things.

Step 1: Updating the nginx.conf file

It is required for the app to not consider SSL. You can rewrite the current nginx.conf with the following file:

upstream toolsapi {
    server api:8000;
}

upstream yolov7 {
    server yolov7:8001;
}

upstream yolov6r1 {
    server yolov6r1:8002;
}

server {
  listen 80;
  listen [::]:80;
  client_max_body_size 1G;
  access_log                /log/access.unsecured.log;
  error_log                 /log/error.unsecured.log error;
  proxy_read_timeout 1h;

  location /yolov7 {
    proxy_pass http://yolov7;
    proxy_redirect off;
    proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
    proxy_set_header Host $http_host;
    proxy_set_header X-Forwarded-Ssl $scheme; 
  }

  location /yolov6r1 {
    proxy_pass http://yolov6r1;
    proxy_redirect off;
    proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
    proxy_set_header Host $http_host;
    proxy_set_header X-Forwarded-Ssl $scheme; 
  }

  location / {
    proxy_pass http://toolsapi;
    proxy_redirect off;
    proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
    proxy_set_header Host $http_host;
    proxy_set_header X-Forwarded-Ssl $scheme;
  }
}

Note: It's better to create a copy of the nginx.conf file before rewriting it.

Step 2: Updating the docker-compose.yml file

The app must be built from local images. You can do it by updating the current docker-compose.yml with the following file:

version: '2'

services:
  api:
    build: .
    ports:
      - 8000:8000
  yolov7:
    build: ./yolov7
    ports:
      - 8001:8001
  yolov6r1:
    build: ./yolov6r1
    ports:
      - 8002:8002
  nginx:
    build: ./nginx
    ports:
      - 80:80
      - 443:443
    depends_on:
      - api
      - yolov7
      - yolov6r1
    volumes:
      - ./log:/log:z
      - /etc/letsencrypt/live/tools.luxonis.com/privkey.pem:/ssl/key.pem
      - /etc/letsencrypt/live/tools.luxonis.com/fullchain.pem:/ssl/cert.pem

Note: It's better to create a copy of the docker-compose.yml file before rewriting it.

Step 3: Building

sudo docker-compose build

Step 4: Running

sudo docker-compose up

Step 5: Open browser

Open browser at http://0.0.0.0/

Credits

This application uses source code of the following repositories: YOLOv5, YOLOv6, YOLOv7, and Ultralytics YOLOv8 (see each of them for more information).

License

This application is available under AGPL-3.0 License license (see LICENSE file for details).