Giter VIP home page Giter VIP logo

bigdatasciencegroup / videorecognition-realtime-autotrainer-alerts Goto Github PK

View Code? Open in Web Editor NEW

This project forked from nasdin/videorecognition-realtime-autotrainer-alerts

1.0 2.0 0.0 43.22 MB

State of the art object detection in real-time using YOLOV2 algorithm. Augmented with a process that alows easy training of the classifier as a plug & play solution . Provides alert if an item in an alert list is detected.

Home Page: http://nasrudinsalim.com

License: GNU General Public License v3.0

Python 100.00%

videorecognition-realtime-autotrainer-alerts's Introduction

Real-Time Video Recognition AI with auto-webscraping trainer + alerts

By: Nasrudin Salim

Real-time object detection and classification with Yolo9000/Yolov2 algorithm

Paper 1 , Paper 2 darkflow darknet

Test:

What is this:

  • Automatically trains a class via webscraping image search results on a video recognition classifier with transfer learning.
  • Enter a label, then enter a list of search queries to google for. It will then google for those search terms and fine-tunes a pretrained classifier.
  • Detect objects as well as output alerts if an object in your "alert list" is found.
  • Can be performed on a video stream in real-time.
  • Can be performed on a live-camera stream in real-time.

Prerequisities

  1. Python 3
  2. ffmpeg
  3. OpenCV
  4. OpenCV-Python
  5. Tensorflow-GPU
  6. CUDNN and CUDA ToolKit

Usage

Getting Data

On the webscraper, indicate the label, as well as the search terms to use by editing it and change the parameters in downloadimages.py then in batch, type

python downloadimages.py

Training

Once images are downloaded. You can download pretrained weights here: darknet Or you can continue training your weights if you've done this before Edit the parameters in train.py and then in batch type:

python train.py

Testing/Using

You can test on either a video file, through video_run.py You can test on a live webcam feed through camera_run.py

python video_run.py
python camera_run.py

You will be asked for the path of the video file. You can adjust the parameters as well as the paths of the weights by opening up the py files.

Testing/Using

You can test on either a video file, through video_run.py You can test on a live webcam beed through camera_run.py

python video_run.py
python camera_run.py

You will be asked for the path of the video file. You can adjust the parameters as well as the paths of the weights by opening up the py files.

Setting Alerts

You can set alerts by editing the text file "alerts.txt" when a label found in this text file appears, it will generate an alert by drawing the box red and displaying "Alert x found in footage" when testing.

videorecognition-realtime-autotrainer-alerts's People

Contributors

nasdin avatar

Stargazers

Lorraine David avatar

Watchers

James Cloos avatar paper2code - bot 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.