Skip to content

Java implementation for Classification and Object-Detection with TensorFlow

License

Notifications You must be signed in to change notification settings

HapsHapsHaps/machinelearning-subs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Object Detection with TensorFlow and Java

I've gone trough a lot of trouble for figuring out how to both prepare a dataset, train a model with TensorFlow, and how to use it from a Java program.
So from all my effort, this repository contains somewhat detailed guides on how you can train your own model.

With this I've located and modified a few scripts to make th training simpler. I've also created customized Docker containers that is based on the official TensorFlow containers, but comes with a bunch of fixes and the needed research project files. Without these, it simply not possible to get any of this stuff to work properly.
I've also created an easy to use java library, that you can use from your own applications.

By packing it all into new Docker containers, everything works the same every time, and it's a lot easier to train the models this way.

I've also created an easy to use java library, that you can use from your own applications.

This entire project was done to make it possible to detect and locate the exact positions of subtitles in videos.

You can find the fairly detailed guide on how to prepare a dataset and train a model for Object Detection in the training folder.

Information about the customized containers, can be found in the folder named docker.

About

Java implementation for Classification and Object-Detection with TensorFlow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published