How to Train a Custom Model for Object Detection (Local and Google Colab!)

This video goes over how to train your custom model using either your local computer, or by utilizing the free GPUs on Google Colab. I also make sure to go over how to fix errors such as ‘no module named nets/deployment’ as I see people trip on that one often. As I said in the video, support for object detection with TensorFlow 2 just recently dropped so expect an update video at some time in the future. TensorFlow has stated that they will keep support for tf1 so don’t worry if you want to keep following this tutorial.

Commands Used:
o python train.py --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config --logtostderr
o python export_inference_graph.py --input_type image_tensor --pipeline_config_path training/ssd_mobilenet_v1_pets.config --trained_checkpoint_prefix training/model.ckpt-3207 --output_directory new_graph

Useful Links:

My GitHub: https://github.com/Bengemon825/TF_Object_Detection2020
TensorFlow Models: http://bit.ly/2McixuB

🔔 Subscribe: https://www.youtube.com/channel/UCoNg3M1usgTsLEXPZ4IzrlA

#tensorflow #python #machine-learning