Skip to content

An easy project to classify image with tensorflow using inception v3

Notifications You must be signed in to change notification settings

xblaster/tensor-guess

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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) You can use pwd as a workaround

About

An easy project to classify image with tensorflow using inception v3

Resources

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published