OpenCV
4.0.0
Open Source Computer Vision
|
Halide code is the same for every device we use. But for achieving the satisfied efficiency we should schedule computations properly. In this tutorial we describe the ways to schedule your networks using Halide backend in OpenCV deep learning module.
For better understanding of Halide scheduling you might want to read tutorials @ http://halide-lang.org/tutorials.
If it's your first meeting with Halide in OpenCV, we recommend to start from How to enable Halide backend for improve efficiency.
You can schedule computations of Halide pipeline by writing textual configuration files. It means that you can easily vectorize, parallelize and manage loops order of layers computation. Pass path to file with scheduling directives for specific device into cv::dnn::Net::setHalideScheduler
before the first cv::dnn::Net::forward
call.
Scheduling configuration files represented as YAML files where each node is a scheduled function or a scheduling directive.
Considered use variables n
for batch dimension, c
for channels, y
for rows and x
for columns. For variables after split are used names with the same prefix but o
and i
suffixes for outer and inner variables correspondingly. In example, for variable x
in range [0, 10)
directive split: { x: 2 }
gives new ones xo
in range [0, 5)
and xi
in range [0, 2)
. Variable name x
is no longer available in the same scheduling node.
You can find scheduling examples at opencv_extra/testdata/dnn and use it for schedule your networks.
Thanks to layers fusing we can schedule only the top layers of fused sets. Because for every output value we use the fused formula. In example, if you have three layers Convolution + Scale + ReLU one by one,
fused function is something like
So only function called relu
require scheduling.
Sometimes networks built using blocked structure that means some layer are identical or quite similar. If you want to apply the same scheduling for different layers accurate to tiling or vectorization factors, define scheduling patterns in section patterns
at the beginning of scheduling file. Also, your patters may use some parametric variables.
You can let DNN to schedule layers automatically. Just skip call of cv::dnn::Net::setHalideScheduler
. Sometimes it might be even more efficient than manual scheduling. But if specific layers require be scheduled manually, you would be able to mix both manual and automatic scheduling ways. Write scheduling file and skip layers that you want to be scheduled automatically.