AI-UI-GmbH / AI-Lab

AI-Lab is a toolbox open sourced by AI-UI GmbH.

Home Page:https://ai-ui.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AI-Lab

AI-Lab is a toolbox open sourced by Team AI-UI for image segmentation and reinforcement learning (later). AI-Lab is built upon tensorflow 2.3.

Now, AI-Lab implements:

  • BlendMask

What's Next

  • Mask R-CNN
  • Train and test on COCO dataset

Quick Start

Change data folder and annotation folder in configs/blendmask.yaml

If not using our software AI-UI or VIA as annotation tool, rewrite load_mask in dataset.py to adapt your annotation.

Run:

python train.py

Train COCO dataset

Download COCO dataset and annotation files here

Change data folder and annotation folder in configs/blendmask.yaml

Run:

python train_coco.py

About

AI-Lab is a toolbox open sourced by AI-UI GmbH.

https://ai-ui.ai

License:MIT License


Languages

Language:Python 100.0%