masc-it/Custom-Object-detector

Tensorflow 2.2: Custom Object detector

Object Detection with Tensorflow 2

Dataset

• Get images (train, test)
• resize images
• LabelImg
• xml_to_csv
• Upload to Kaggle

Installation, Training & export (Colab)

https://colab.research.google.com/github/TannerGilbert/Tensorflow-Object-Detection-API-Train-Model/blob/master/Tensorflow_2_Object_Detection_Train_model.ipynb

Local dependencies setup

Same as colab https://www.youtube.com/watch?v=cvyDYdI2nEI

• protoc: https://github.com/protocolbuffers/protobuf/releases/download/v3.13.0/protoc-3.13.0-win64.zip

Test 1

Live detection from local webcam

python detect_from_webcam.py -m <path_to_model_dir> -l <complete_pbtxt_path>

Test 2

Live detection from local webcam, with webserver (mobile, desktop)

python server.py -m <path_to_model> -l <complete_pbtxt_path>