#!/bin/bash

pushd "$(dirname $0)"/../.. > /dev/null
pushd resource > /dev/null

# build converter
CONVERTER=../build/MNNConvert
if [ ! -e ${CONVERTER} ]; then
  echo "can't find ${CONVERTER}, building converter firstly "
  exit
fi

# functions
download() {
  if [ -e $2 ]; then return 0; fi

  name=`basename $2`
  echo "downloading $name ..."
  status=`curl $1 -s -w %{http_code} -o $2`
  if (( status == 200 )); then
    return 0
  else
    echo "download $name failed" 1>&2
    return -1
  fi
}

get_caffe1() { # model_URL, model_path, prototxt_URL, prototxt_path, model, MNN_path
  if [ ! -e $6 ]; then
    echo "download and convert $2 $4"
    download $1 $2 && download $3 $4 && ./$CONVERTER -f CAFFE --modelFile $2 --prototxt $4 --MNNModel $6 --bizCode 0000 --keepInputFormat=0
  fi
}

get_tensorflow_lite() {
  if [ ! -e $4 ]; then
    mkdir -p build
    pushd build > /dev/null
    download $1 $2.tgz && tar -xzf $2.tgz $2
    succ=$?
    popd > /dev/null
    [ $succ -eq 0 ] && ./$CONVERTER -f TFLITE --modelFile build/$2 --MNNModel $4 --bizCode 0000 --keepInputFormat=0
  fi
}

get_portrait_lite() {
  if [ ! -e $4 ]; then
    mkdir -p build
    pushd build > /dev/null
    download $1 $2
    succ=$?
    popd > /dev/null
    [ $succ -eq 0 ] && ./$CONVERTER -f TFLITE --modelFile build/$2 --MNNModel $4 --bizCode 0000 --keepInputFormat=0
  fi
}

# get models

## Using MobileNet V1 downloaded from: https://github.com/shicai/MobileNet-Caffe/
get_caffe1 \
  "https://raw.githubusercontent.com/shicai/MobileNet-Caffe/master/mobilenet.caffemodel" \
  "build/mobilenet_v1.caffe.caffemodel" \
  "https://raw.githubusercontent.com/shicai/MobileNet-Caffe/master/mobilenet_deploy.prototxt" \
  "build/mobilenet_v1.caffe.prototxt" \
  "MobileNet V1" \
  "model/MobileNet/v1/mobilenet_v1.caffe.mnn"

## Using MobileNet V2 downloaded from: https://github.com/shicai/MobileNet-Caffe/
get_caffe1 \
  "https://raw.githubusercontent.com/shicai/MobileNet-Caffe/master/mobilenet_v2.caffemodel" \
  "build/mobilenet_v2.caffe.caffemodel" \
  "https://raw.githubusercontent.com/shicai/MobileNet-Caffe/master/mobilenet_v2_deploy.prototxt" \
  "build/mobilenet_v2.caffe.prototxt" \
  "MobileNet V2" \
  "model/MobileNet/v2/mobilenet_v2.caffe.mnn"

## Using SqueezeNet V1.0 downloaded from: https://github.com/DeepScale/SqueezeNet/
get_caffe1 \
  "https://raw.githubusercontent.com/DeepScale/SqueezeNet/master/SqueezeNet_v1.0/squeezenet_v1.0.caffemodel" \
  "build/squeezenet_v1.0.caffe.caffemodel" \
  "https://raw.githubusercontent.com/DeepScale/SqueezeNet/master/SqueezeNet_v1.0/deploy.prototxt" \
  "build/squeezenet_v1.0.caffe.prototxt" \
  "SqueezeNet V1.0" \
  "model/SqueezeNet/v1.0/squeezenet_v1.0.caffe.mnn"

## Using SqueezeNet V1.1 downloaded from: https://github.com/DeepScale/SqueezeNet/
get_caffe1 \
  "https://raw.githubusercontent.com/DeepScale/SqueezeNet/master/SqueezeNet_v1.1/squeezenet_v1.1.caffemodel" \
  "build/squeezenet_v1.1.caffe.caffemodel" \
  "https://raw.githubusercontent.com/DeepScale/SqueezeNet/b6b5ae2ce884a3866c21efd31e103defde8631ae/SqueezeNet_v1.1/deploy.prototxt" \
  "build/squeezenet_v1.1.caffe.prototxt" \
  "SqueezeNet V1.1" \
  "model/SqueezeNet/v1.1/squeezenet_v1.1.caffe.mnn"

## Using MobileNet V2 downloaded from: http://download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224.tgz
get_tensorflow_lite \
  "http://download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224.tgz" \
  "mobilenet_v2_1.0_224.tflite" \
  "MobileNet V2 TFLite" \
  "model/MobileNet/v2/mobilenet_v2_1.0_224.tflite.mnn"

## Using MobileNet V2 downloaded from: http://download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224_quant.tgz
get_tensorflow_lite \
  "http://download.tensorflow.org/models/tflite_11_05_08/mobilenet_v2_1.0_224_quant.tgz" \
  "mobilenet_v2_1.0_224_quant.tflite" \
  "MobileNet V2 TFLite Quantized" \
  "model/MobileNet/v2/mobilenet_v2_1.0_224_quant.tflite.mnn"

## Using deeplab v3 downloaded from: https://storage.googleapis.com/download.tensorflow.org/models/tflite/gpu/deeplabv3_257_mv_gpu.tflite
get_portrait_lite \
  "https://storage.googleapis.com/download.tensorflow.org/models/tflite/gpu/deeplabv3_257_mv_gpu.tflite" \
  "deeplabv3_257_mv_gpu.tflite" \
  "deeplabv3" \
  "model/Portrait/Portrait.tflite.mnn"

popd > /dev/null
popd > /dev/null