Train and predict an object detection model on docker using tensorflow object detection 2.0
Steps to follow:
- To build image, update the root directory of tensorflow and the name of the image in the
Build.bat
file. - Copy the Dockerfile from this repository to
.\research\object_detection\dockerfiles\tf2\Dockerfile
location. - Run the
Build.bat
file. - Clone this repo and create a folder name
images
and paste there all the training images along with annotations (xml) files. - Update the path of this repository directory in
Run.bat
file and confirm the ports as per the availability. - Run the
Run.bat
file to run docker image. - Follow the
Final_OD_pothole.ipynb
file to train and predict the object detection model.