jerryc05 / Hello-Object-Detection

This is a tutorial to help you setup Object Detection using TensorFlow Object Detection API.

Hello Object Detection

This is a tutorial to help you setup Object Detection using TensorFlow Object Detection API.

Author: @jerryc05

For latest setup process, please refer to the following sites:

• tensorflow/models/object_detection
• tensorflow/models

Table of Contents:

1. Installation
   1. Setup Tensorflow-GPU
   2. Setup PyPi Libraries
   3. Setup COCO API
   4. Install Protocol Buffer
   5. Compile protobuf models and parameters
   6. Add Libraries to PYTHONPATH
   7. Test Installation
2. Label Image
   1. Install labelImg
   2. Use labelImg
   3. Parse labels
3. Start Training
   1. Configure Training
   2. Run Training
4. Freeze Model

Installation

For latest setup process, please refer to the following site:

• tensorflow/models/object_detection/g3doc/installation.md

Setup Tensorflow-GPU

Please refer to the official tutorials:

• TensorFlow GPU Support
• TensorFlow GPU Support (Mainland China)

Setup PyPi Libraries

Run the following command in shell from folder ./:

python -m pip install --user -r requirements.txt

Setup COCO API

For latest setup process, please refer to the official site cocodataset/cocoapi.

Windows or Unix users:

1. Download cocodataset/cocoapi/PythonAPI and cocodataset/cocoapi/common folders.

2. Extract PyhonAPI and common folder so that the directory looks like:

   Hello-Object-Detection
   |-- cocoapi
   |   |-- common
   |       |-- maskApi.c
   |       |-- ...
   |   |-- PythonAPI
   |       |-- setup.py
   |       |-- ...
   |-- main.py
   |-- ...
   

3. IMPORTANT! Modify the file ./cocoapi/PythonAPI/setup.py as follow:

   • Delete the line that contains extra_compile_args.

4. Build and install pycocotools:

   Run the following command in shell from folder ./cocoapi/PythonAPI:

   python setup.py install

5. Successful output will output Finished processing dependencies for pycocotools==... as the last line of output.

Unix users only:

Run the following command in shell from folder ./:

python -m pip install pycocotools

Install Protocol Buffer

1. Download the corresponding binary zipped release from protocolbuffers/protobuf.
2. Unzip the folder.
3. Add protoc executable to Environment Variable PATH then restart terminal. (Or you will need to replace every protoc call with PATH_TO_PROTOC/protoc later.)

Compile protobuf models and parameters

1. Download tensorflow/models/research/object_detection and tensorflow/models/research/slim folders.

2. Extract object_detection and slim folders so that the directory looks like:

   Hello-Object-Detection
   |-- models
   |   |-- research
   |       |-- object_detection
   |           |-- model_main.py
   |           |-- ...
   |       |-- slim
   |           |-- setup.py
   |           |-- ...
   |-- main.py
   |-- ...
   

3. Run the following command in shell from folder ./models/research:

   protoc object_detection/protos/*.proto --python_out=.

4. Successful execution will output nothing.

Add Libraries to PYTHONPATH

The following path shall be appended to Environment Variable PYTHONPATH:

1. __PWD__/models/research
2. __PWD__/models/research/slim

Note: change __PWD__ to the absolute path of Hello-Object-Detection folder before touching PYTHONPATH!

Test Installation

1. Run the following command in shell from folder ./models/research:

   python object_detection/builders/model_builder_test.py

2. Successful execution will output OK or OK (skipped=...) as the last line of output.

Label Image

For latest setup process, please refer to site: tzutalin/labelImg

Install labelImg

Run the following command in shell from folder ./:

python -m pip install labelImg

Use labelImg

No Bullshit here, please refer to official site tzutalin/labelImg.

Parse labels

1. Parse labels from xml to csv:

   Run the following command in shell from folder ./:

   python utils/xml_to_csv.py -i ${PATH_TO_XML_FOLDER} -o ${PATH_TO_CSV_FOLDER}

   Note: change the following paths before running the script:

   1. ${PATH_TO_XML_FOLDER} <- path of xml labels created by labelImg.
   2. ${PATH_TO_CSV_FOLDER} <- path where csv file will be saved.

2. Parse csv to TFRecord:

   Run the following command in shell from folder ./:

   python utils/csv_to_tfrecord.py -c ${PATH_TO_CSV_FILE} -i ${PATH_TO_IMG_FOLDER} -o ${PATH_TO_TFRECORD_FILE_FOLDER}

   Note: change the following paths before running the script:

   1. ${PATH_TO_CSV_FILE} <- path of csv file created by xml_to_csv.py.
   2. ${PATH_TO_IMG_FOLDER} <- path of image files.
   3. ${PATH_TO_TFRECORD_FILE_FOLDER} <- path of TFRecord file that the script will create.

   Note: you might need to run it twice for both train.csv and eval.csv respectively.

Start Training

Configure Training

1. Select a pre-configure model config from ./models/research/object_detection/samples/configs.

2. Copy the .config file to somewhere else.

   For example, I use ./config/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config.

   So the directory looks like:

   Hello-Object-Detection
   |-- training
   |   |-- ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config
   |-- main.py
   |-- ...
   

3. Open the config file.

4. Replace the value of num_classes with your number of classes.

   If you forgot the number, check your .pbtxt file that contains label map.

5. Replace the value of fine_tune_checkpoint with the path of model checkpoint file to save.

   For example, I use "./config/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.ckpt".

   IMPORTANT! If the pre-trained model does not exist yet, comment out this line by adding a # at the begining.

6. Replace the value of num_steps with the number of steps to train.

   For example, I use "./config/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.ckpt".

   Note: If the model should be trained indefinitely, comment out this line by adding a # at the beginning.

7. Replace the value of input_path under section train_input_reader with the path of training TFRecord file.

   Note: wildcard char ? can be used to match multiple file names.

8. Replace the value of label_map_path under section train_input_reader with the path of training label map .pbtxtfile.

9. OPTIONAL: Replace the value of num_examples with the number of images to eval.

10. Replace the value of input_path under section eval_input_reader with the path of eval TFRecord file.

    Note: wildcard char ? can be used to match multiple file names.

11. Replace the value of label_map_path under section eval_input_reader with the path of eval label map .pbtxtfile.

Run Training

Run the following command in shell from folder ./:

python models/research/object_detection/model_main.py --pipeline_config_path=${PATH_TO_CONFIG_FILE} --model_dir=${PATH_TO_CHECKPOINT_FOLDER} --alsologtostderr

Note: change the following paths before running the script:

1. ${PATH_TO_CONFIG_FILE} <- path of pre-configured model config file.
2. ${PATH_TO_CHECKPOINT_FOLDER} <- path where training checkpoints and events will be saved.

Freeze Model

Run the following command in shell from folder ./:

python models/research/object_detection/export_inference_graph.py --pipeline_config_path ${PIPELINE_CONFIG_PATH} --output_directory ${OUTPUT_PATH} --trained_checkpoint_prefix ${CKPT_PATH_PREFIX_ONLY}

Note: change the following paths before running the script:

1. ${PIPELINE_CONFIG_PATH} <- path of pre-configured model config file.
2. ${OUTPUT_PATH} <- path where the frozen model will be saved.
3. ${CKPT_PATH_PREFIX_ONLY} <- path of the checkpoint file, discarding postfixes such as .meta, .index, and .data-....