OBSEA Fish Detection

AI-based fish detection and classification algorithm based on YOLOv8. The model has been finetuned to detect and classify fish at the OBSEA underwater observatory.

This is a container that will run the obsea-fish-detection application leveraging the DEEP as a Service API component (DEEPaaS API). The application is based on ai4oshub/ai4os-yolov8-torch module.

Running the container

Directly from Docker Hub

To run the Docker container directly from Docker Hub and start using the API, simply run the following command:

$ docker run -ti -p 5000:5000 -p 6006:6006 -p 8888:8888 ai4oshub/obsea-fish-detection

This command will pull the Docker container from the Docker Hub ai4oshub repository and start the default command (deepaas-run --listen-ip=0.0.0.0).

N.B. For either CPU-based or GPU-based images you can also use udocker.

Building the container

If you want to build the container directly in your machine (because you want to modify the Dockerfile for instance) follow the following instructions:

git clone https://github.com/ai4os-hub/obsea-fish-detection
cd obsea-fish-detection
docker build -t ai4oshub/obsea-fish-detection .
docker run -ti -p 5000:5000 -p 6006:6006 -p 8888:8888 ai4oshub/obsea-fish-detection

These three steps will download the repository from GitHub and will build the Docker container locally on your machine. You can inspect and modify the Dockerfile in order to check what is going on. For instance, you can pass the --debug=True flag to the deepaas-run command, in order to enable the debug mode.

Connect to the API

Once the container is up and running, browse to http://0.0.0.0:5000/ui to get the OpenAPI (Swagger) documentation page.

Project structure

├─ Dockerfile             <- Describes main steps on integration of DEEPaaS API and
│                            <your_project> application in one Docker image
│
├─ Jenkinsfile            <- Describes basic Jenkins CI/CD pipeline
│
├─ LICENSE                <- License file
│
├─ README.md              <- README for developers and users
│
├── .sqa/                 <- CI/CD configuration files
│
└─ metadata.json          <- Defines information propagated to the AI4OS Hub

You can validate the metadata.json before making a git push using:

pip install ai4-metadata-validator
ai4-metadata-validator metadata.json