model.yml

# Copyright (c) 2020 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: >-
  FaceBoxes: A CPU Real-time Face Detector with High Accuracy. For details see the
  repository <https://github.com/zisianw/FaceBoxes.PyTorch>, paper <https://arxiv.org/pdf/1708.05234.pdf>
task_type: detection
files:
  - name: faceboxes.py
    size: 4999
    sha256: dadc02f4324565999f19affcdb26456b854b6ddbcba3892262fbee18b783c874
    source: https://raw.githubusercontent.com/zisianw/FaceBoxes.PyTorch/ab8478c22627690899114950cae27f8093424c95/models/faceboxes.py
  - name: FaceBoxesProd.pth
    size: 4072492
    sha256: c0cb2b1e40710c0aa5fc32a8759b1496a0fe0a126c907ca0ffe35b4bd0709d09
    source:
      $type: google_drive
      id: 1tRVwOlu0QtjvADQ2H7vqrRwsWEmaqioI
conversion_to_onnx_args:
  - --model-path=$dl_dir
  - --model-name=FaceBoxes
  - --import-module=faceboxes
  - --model-param=phase='test'
  - --model-param=size=None
  - --model-param=num_classes=2
  - --weights=$dl_dir/FaceBoxesProd.pth
  - --input-shape=1,3,1024,1024
  - --input-names=input.1
  - --output-names=boxes,scores
  - --output-file=$conv_dir/faceboxes-pytorch.onnx
model_optimizer_args:
  - --input_shape=[1,3,1024,1024]
  - --input=input.1
  - --input_model=$conv_dir/faceboxes-pytorch.onnx
  - --mean_values=input.1[104.0,117.0,123.0]
  - --output=boxes,scores
framework: pytorch
license: https://github.com/zisianw/FaceBoxes.PyTorch/blob/master/LICENSE