Git Product home page Git Product logo

detection's Introduction

AI-Deck Object Detection

(http://url/to/img.png)

Repo contains:

  1. Training a simple object detection model (MobileNetv2)
  2. Quantising TFLite Model
  3. Generating Autotiler Code for AI-Deck implementation

Quickstart

  1. Clone repository into aideck-gap8-examples/examples/ai/ folder
  2. From a terminal with the docker container, or gap_sdk dev environment, in the aideck-gap8-examples/ folder, execute:
    • docker run --rm -v ${PWD}:/module aideck-with-autotiler tools/build/make-example examples/ai/detection clean model build image
  3. From another terminal (outside of the container), use the cfloader to flash the example if you have the gap8 bootloader flashed AIdeck. Change the [CRAZYFLIE URI] with your crazyflie URI like radio://0/90/2M/E7E7E7E726.
    • cfloader flash examples/ai/detection/BUILD/GAP8_V2/GCC_RISCV_FREERTOS/target.board.devices.flash.img deck-bcAI:gap8-fw -w radio://0/90/2M/E7E7E7E726
    • Flashing will take ~10 minutes depending on the model size and it may appear to be stuck at 99%. However, if the crazyflie automatically reboots at 99%, the flashing is usually completed and you can Ctrl+C in the terminal.
    • Reboot the Crazyflie.

When the example is flashing, you should see the GAP8 LED blink fast, which is the bootloader. The example itself can be noticed by a slow blinking LED. You should also receive the detection output in the cfclient console.

Custom Model

Dataset

  1. Download dataset from Roboflow in Pascal VOC format.
  2. Extract them in aideck-gap8-examples/examples/ai/detection/images/.
  3. Run python3 pascal_csv.py.

Training

  • Refer to model.ipynb to create detection_q.tflite model. Make sure tflite model is saved in model/ folder.

detection's People

Contributors

cweekiat avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.