1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

#!/bin/bash

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

set -e

if [ ! -d "./models" ]( ! -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" ]( ! -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 完毕。"