#!/bin/bash

# Description: 快速更新至最新 tensorflow-models 代码
# Author: liujun
# Date: 2019-05-28

set -e

if [[ ! -d "./models" ]]; then
    # 如果未存在 models ，则先下载
    echo "== 01 == 未找到 models 目录，准备下载。"
    git clone https://github.com/tensorflow/models.git
    echo "== 02 == 下载 models 完毕。"
else
    # 还原并更新代码
    echo "== 01 == 还原并更新 models 代码开始。"
    cd models
    git reset --hard
    git clean -xdf
    git checkout master
    git pull
    cd ..
    echo "== 02 == 还原并更新 models 代码完毕。"
fi

# 如果未存在 cocoapi ，则先下载
if [[ ! -d "./cocoapi" ]]; then
    echo "== 03 == 未找到 cocoapi 目录，准备下载。"
    git clone https://github.com/cocodataset/cocoapi.git
    echo "== 04 == 下载 cocoapi 完毕。"
else
    # 还原并更新代码
    echo "== 03 == 还原并更新 cocoapi 代码开始。"
    cd cocoapi
    git reset --hard
    git clean -xdf
    git checkout master
    git pull
    cd ..
    echo "== 04 == 还原并更新 cocoapi 代码完毕。"
fi

# 编译 cocoapi 并拷贝到 models 中。
echo "== 05 == 准备编译 cocoapi 。"
cd ./cocoapi/PythonAPI
# make
python3 setup.py build_ext --inplace
rm -rf build
cd ../../
echo "== 06 == 编译 cocoapi 完毕。"
rm -rf ./models/research/pycocotools/
cp -a ./cocoapi/PythonAPI/pycocotools ./models/research/pycocotools
echo "== 07 == 已拷贝 cocoapi 至 models 中。"

# 编译 models object detection 。
echo "== 08 == 准备编译 object detection 。"
cd ./models/research
protoc ./object_detection/protos/*.proto --python_out=.
cd ../../
echo "== 09 == 编译 object detection 完毕。"