Skip to main content

Quick detection practice for images and videos using TensorFlow

Project description

TFlash

A quick way to practice object detection on images and videos for common classes using TensorFlow. Fully operational by any user without Machine Learning experience.

Requirements

imageio
imageio-ffmpeg (for videos)
tensorflow
tqdm

Installation

pip install tflash

Note: due to the incompatibility of numpy versions, in some cases tensorflow needs to be (re-)installed after installing imageio

Usage

import tflash

flash = tflash.Detector()
result = flash.detect("a_pic.jpg", print_output="a_result.jpg") # can be set to False
result = flash.detect("mypic.jpg", min_score=0.75) # default: 0.5

my_pics = ["pic001.jpg", "pic002.jpg", "pic004.jpg"]
result = flash.detect_multiple(my_pics)

Output:

  • Dict of detection results
  • Labeled image(s)

detections = result["detections"] # formatted as dict
print("Saved to {}".format(result["output"]) # output filename

Font

Alter font (default is Roboto size 20):
flash.set_font("arial.ttf")
flash.set_font_size(12)
flash.set_font("dosis.ttf", size=15)

Own Model File

Some good ones are provided at Tensorflow model zoo: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md

Download a file with output "Boxes" from the link above, extract and use only the one with *.pb extension, e.g., frozen_inference_graph.pb

Load in TFlash using the command:
flash.load_model("frozen_inference_graph.pb")
or
flash = tflash.Detector("any_other_model_file.pb")

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tflash-0.5.1.tar.gz (101.6 kB view hashes)

Uploaded Source

Built Distribution

tflash-0.5.1-py3-none-any.whl (100.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page