This repo contains scripts and tutorials for object detection models training.
- Face Detection - models that are able to detect faces.
- Horizontal Text Detection - model that is able to detect more or less horizontal text with high speed.
- Person Detection - models that are able to detect persons.
- Person Vehicle Bike Detection - models that are able to detect 3 classes of objects: person, vehicle, non-vehicle (e.g. bike).
- Vehicle Detection - models that are able to detect vehicles.
- Ubuntu* 16.04
- Python* >=3.6
- PyTorch* 1.5.1
- OpenVINO™ 2020.4 with Python API
- mmdetection (../../external/mmdetection)
- Create virtual environment and build mmdetection:
bash init_venv.sh
- Activate virtual environment:
. venv/bin/activate