/TFOD--BeginnerLevel-Project

A beginner-level Tensorflow Object Detection Project using tf v1.14.

Primary LanguageJupyter Notebook

TFOD--BeginnerLevel-Project

A beginner-level Tensorflow Object Detection Project using tf v1.14.
Credit : The codes(with some local changes) are from the original repo : https://github.com/tensorflow/models/tree/v1.13.0
The image annotation tool 'LabelImg can be obtained from its original source : https://github.com/tzutalin/labelImg


The model architecture used for this project is "faster_rcnn_inception_v2_coco".
The model was trained locally for initial 234 epochs on "i7-8750H CPU @ 2.20GHz" + 8GB DDR4 RAM using tensorflow V1.14. For further improvement of model, it was trained on Colab till a total of 1000 epochs with GPU enabled.
The steps followed to train the model has been mentioned in the 'Read Me (Steps)' file in order to revise the steps or recreate the similar folder structure on future projects.
The Colab notebook is also in the repo with training outputs.
It has been observed that after 1000 epochs of training its able to identify cards that are completely in view. In case of partially concealed cards, its able to make a few correct predictions. Perhaps a 1000 more epochs will enable even better prediction(an experiment for next github commit).

Note : The file "models\research\training\model.ckpt-1000.data-00000-of-00001" has been zipped to commit as it was over 100MB.