model.yml

# Copyright (c) 2019 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

description: >-
  The `mtcnn-r` model is one of the mtcnn <https://arxiv.org/ftp/arxiv/papers/1604/1604.02878.pdf>
  group of models designed to perform face detection. Short for "Multi-task Cascaded
  Convolutional Neural Network", it is implemented using the Caffe\* framework.
  The "r" designation indicates that this model is the "refine" network intended
  to refine the data returned as output from the "proposal" `mtcnn-p` network.
  For details about this family of models, check out the repository <https://github.com/DuinoDu/mtcnn>.

  The model input is a blob with a vector containing the first pass of face data,
  as returned by the `mtcnn-p` model. The mean values need to be subtracted as
  follows: [127.5, 127.5, 127.5] before passing the image blob into the network.
  In addition, values must be divided by 0.0078125.

  The model output is a blob with a vector containing the refined face data. If
  there are no faces detected by the refine pass, no further processing is needed.
  Otherwise, you will typically use this output as input to the `mtcnn-o` model.
task_type: detection
files:
  - name: mtcnn-r.prototxt
    size: 3159
    sha256: 077686e89e606354f425366afdb2018777d93c6450b50e2c12301f8a97f6bb47
    source: https://raw.githubusercontent.com/DuinoDu/mtcnn/db5bd8f02023f8d37913140fd2bf2749c2dbf266/model/det2.prototxt
  - name: mtcnn-r.caffemodel
    size: 407910
    sha256: 39b20f7a57bb8176cc9466cea4dfd52da6a6f876de60c7ab222a309f2d0ca08c
    source: https://github.com/DuinoDu/mtcnn/raw/db5bd8f02023f8d37913140fd2bf2749c2dbf266/model/det2.caffemodel
model_optimizer_args:
  - --input_shape=[1,3,24,24]
  - --input=data
  - --mean_values=data[127.5,127.5,127.5]
  - --scale_values=data[128.0]
  - --output=conv5-2,prob1
  - --input_model=$dl_dir/mtcnn-r.caffemodel
  - --input_proto=$dl_dir/mtcnn-r.prototxt
framework: caffe
license: https://raw.githubusercontent.com/DuinoDu/mtcnn/master/LICENSE