nearkyh / RealTimeShipDetector

:ship: A Real-time ship detector using Tensorflow Object Detection API and OpenCV.

Real-Time Ship Detector

A Real-time ship detector using Tensorflow Object Detection API and OpenCV.

Requirements

• Ubuntu 16.04
• Python 3.5
• TensorFlow-GPU 1.9
• CUDA 9.0
• cuDNN 7.1.4
• OpenCV 3.4

Getting Started

Creating virtualenv

$ cd RealTimeShipDetector
$ virtualenv env --python=python3.5
$ source env/bin/activate

Install dependencies

$ pip install -r requirements.txt

Run

$ python detector.py \
    --video=file_path \
    --model=model_name