How to Train a Custom Model for Object Detection (Local and Google Colab!)
Hey welcome back, Ben again! Today's video is the last part of my object detection tutorial series. 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.
NEW: Tensorflow 2 custom model tutorial • Tensorflow 2 Custom Object Detection ...
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_Obj...
sentdex's YouTube: / @sentdex
TensorFlow Models: https://github.com/tensorflow/models/...