UAV-ZJU-TF-GATE-DETECTION

Detecting the gate with Tensorflow-object-detection-API

Dependencies

This code based on the Tensorflow object detection API. you should install this library.

Train

The pretrained model SSD-MobileNet-V2 was used for fineturn.training this model by running this cmd:

# From tensorflow/models/research/object_detection
python model_main.py --logtostderr --train_dir=./outputing --pipeline_config_path=./ssd_mobilenet_v2_coco.config

Test

# From tensorflow/models/research/object_detection
python video_detection.py