Skip to content

Karlen92/transfer_learnig

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

transfer_learnig

The project allows to build a image classifier in short time, by using transfer learning, from Googles Inception_v3 model. It uses tensorflow 1.7.0, pyhton 3.5.0, node JS 8.9 and requires 64 bit architecture os.

To train on your own data sent should be done the following.

  1. Create folder named samples in project directory, in must contain all training data in it.
  2. The training data must be classified by folders, i.e each folder represent exactly one class, the name of files are not restricted.
  3. To start training open any command line terminal and run npm start, it may take up to 1 hour depending on your system and configuration parameters that you provided(parameters will be provided below)
  4. After training is finished you can now use the model by calling image=[image_file_path] node label.js and the results will be printed in your console.
  5. You can also see the training process as well. Type tensorboard --logdir ./ in root directory of project, it will create a web server and provide link, where you can see reports of training process in your web browser.

The initial configuration is as follows and it can be controlled by environment variables or by editing the server.js file.

	'image_dir': env.image_dir || './tmp/data/',
	'how_many_training_steps':env.how_many_training_steps || 500,
	'testing_percentage': env.testing_percentage ||10,
	'validation_percentage': env.validation_percentage ||10,
	'train_batch_size': env.train_batch_size ||100,
	'final_tensor_name': env.final_tensor_name ||'final_result',
	'trainer_file_path':'train.py'

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published