Person, vehicle, bike detection with multiple data sources

The purpose of this demo is to show how to send data from multiple sources (cameras, video files) to a model served in OpenVINO Model Server.

Deploy person, vehicle, bike detection model

Download model files

mkdir -p model/1

wget -P model/1 https://storage.openvinotoolkit.org/repositories/open_model_zoo/2022.1/models_bin/2/person-vehicle-bike-detection-crossroad-0078/FP32/person-vehicle-bike-detection-crossroad-0078.bin

wget -P model/1 https://storage.openvinotoolkit.org/repositories/open_model_zoo/2022.1/models_bin/2/person-vehicle-bike-detection-crossroad-0078/FP32/person-vehicle-bike-detection-crossroad-0078.xml

Run OpenVINO Model Server

docker run -d -v `pwd`/model:/models -p 9000:9000 openvino/model_server:latest --model_path /models --model_name person-vehicle-detection --port 9000 --shape auto

Running the client application

git clone https://github.com/openvinotoolkit/model_server.git
cd model_server/demos/person_vehicle_bike_detection/python
pip3 install -r requirements.txt
python person_vehicle_bike_detection.py --help

Arguments

Argument

Description

-h, help

Show help message and exit

-n NETWORK_NAME, network_name NETWORK_NAME

Network name

-l INPUT_LAYER, input_layer INPUT_LAYER

Input layer name

-o OUTPUT_LAYER, output_layer OUTPUT_LAYER

Output layer name

-d FRAME_SIZE, frame_size FRAME_SIZE

Input frame width and height that matches used model

-c NUM_CAMERAS, num_cameras NUM_CAMERAS

Number of cameras to be used

-f FILE, file FILE

Path to the video file

-i IP, ip IP

IP address of the ovms

-p PORT, port PORT

Port of the ovms

Using with video file

Copy example video file:

git clone "https://github.com/intel-iot-devkit/sample-videos.git"

Set camera count to 0 with -c 0 and provide path to the video file with -f parameter.

python person_vehicle_bike_detection.py -n person-vehicle-detection -l data -o detection_out -d 1024 -c 0 -f sample-videos/person-bicycle-car-detection.mp4 -i localhost -p 9000

Output:

[$(levelname)s ] Video0 fps: 7, Inf fps: 7, dropped fps: 0
[$(levelname)s ] Video0 fps: 7, Inf fps: 7, dropped fps: 0
[$(levelname)s ] Video0 fps: 7, Inf fps: 7, dropped fps: 0

Using with video file and camera

Set camera count to 1 with -c 1 and provide path to the video file with -f parameter.

python person_vehicle_bike_detection.py -n person-vehicle-detection -l data -o detection_out -d 1024 -c 1 -f sample-videos/person-bicycle-car-detection.mp4 -i localhost -p 9000

Console logs:

[$(levelname)s ] Video1 fps: 7, Inf fps: 7, dropped fps: 0
[$(levelname)s ] Camera0 fps: 7, Inf fps: 7, dropped fps: 0
[$(levelname)s ] Video1 fps: 7, Inf fps: 7, dropped fps: 0
[$(levelname)s ] Camera0 fps: 7, Inf fps: 7, dropped fps: 0
[$(levelname)s ] Video1 fps: 7, Inf fps: 7, dropped fps: 0
[$(levelname)s ] Camera0 fps: 8, Inf fps: 8, dropped fps: 0

NOTE: You should also be seeing the GUI showing the video frame and bounding boxes drawn with the detected class name