Public | Automated Build

Last pushed: 2 months ago
Short Description
Docker image to simplify classification of images with tensorflow
Full Description

tensor-guess

this project is just an helper with docker images based on tutorial from https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/index.html

##requirements

  • docker !
  • git clone this project in a directory

##usage

You just need to make a "classifier" directory with a directory "data" inside it with all your images
For example

 [any_path]/my_own_classifier/
 [any_path]/my_own_classifier/data
 [any_path]/my_own_classifier/data/car
 [any_path]/my_own_classifier/data/moto
 [any_path]/my_own_classifier/data/bus

and then put your image on it.
This "classifier" directory will have your samples but also trained classifier after execution of "train.sh".

##Train process

Just type

 ./train.sh [any_path]/my_own_classifier

And it will do anything for you !

##Guess process

Just type for a single guess

 ./guess.sh [any_path]/my_own_classifier /yourfile.jpg

To guess an entire directory

./guessDir.sh [any_path]/classifier [any_path]/srcDir [any_path]/destDir

Example of result

# ./guess.sh /synced/tensor-lib/moto-classifier/ /synced/imagesToTest/moto21.jpg
moto (score = 0.88331)
car (score = 0.11669)

##Remarks

Use absolute file path for classifier and images because for the moment my script do not support relative path (volume mounting)

Docker Pull Command
Owner
xblaster
Source Repository

Comments (0)