Skip to content

SerhatTurann/TensorflowCustomObjectDetection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

TensorflowCustomObjectDetection

daisy_38 dandelion_50 rose_114

tulip_67 magnolia water_lily_201

Run

After the installation and labelling;

Create tfrecord files

cd training_demo python generate_tf_record.py -x images/train -l annotations/label_map.pbtxt -o annotations/train.record

python generate_tf_record.py -x images/test -l annotations/label_map.pbtxt -o annotations/test.record

Train

python model_main.py --alsologtostderr --model_dir=training/ --pipeline_config_path=training/ssd_inception_v2_coco.config

Save model as pb

python export_inference_graph.py --input_type image_tensor --pipeline_config_path training/ssd_inception_v2_coco.config --trained_checkpoint_prefix training/model.ckpt-13302 --output_directory trained-inference-graphs/output_inference_graph_v1.pb

Resources

https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/install.html

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published