File size: 4,336 Bytes
7b7527a |
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 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 |
#!/usr/bin/env bash
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
#
# 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.
#=================================================
# Utils
#=================================================
# directory config
DIST_DIR="dist"
BUILD_DIR="build"
EGG_DIR="paddledet.egg-info"
CFG_DIR="configs"
TEST_DIR=".tests"
DATA_DIR="dataset"
# command line log config
RED='\033[0;31m'
BLUE='\033[0;34m'
GREEN='\033[1;32m'
BOLD='\033[1m'
NONE='\033[0m'
function python_version_check() {
PY_MAIN_VERSION=`python -V 2>&1 | awk '{print $2}' | awk -F '.' '{print $1}'`
PY_SUB_VERSION=`python -V 2>&1 | awk '{print $2}' | awk -F '.' '{print $2}'`
echo -e "find python version ${PY_MAIN_VERSION}.${PY_SUB_VERSION}"
if [ $PY_MAIN_VERSION -ne "3" -o $PY_SUB_VERSION -lt "5" ]; then
echo -e "${RED}FAIL:${NONE} please use Python >= 3.5 !"
exit 1
fi
}
function init() {
echo -e "${BLUE}[init]${NONE} removing building directory..."
rm -rf $DIST_DIR $BUILD_DIR $EGG_DIR $TEST_DIR
if [ `pip list | grep paddledet | wc -l` -gt 0 ]; then
echo -e "${BLUE}[init]${NONE} uninstalling paddledet..."
pip uninstall -y paddledet
fi
echo -e "${BLUE}[init]${NONE} ${GREEN}init success\n"
}
function build_and_install() {
echo -e "${BLUE}[build]${NONE} building paddledet wheel..."
python setup.py sdist bdist_wheel
if [ $? -ne 0 ]; then
echo -e "${RED}[FAIL]${NONE} build paddledet wheel failed !"
exit 1
fi
echo -e "${BLUE}[build]${NONE} ${GREEN}build paddldet wheel success\n"
echo -e "${BLUE}[install]${NONE} installing paddledet..."
cd $DIST_DIR
find . -name "paddledet*.whl" | xargs pip install
if [ $? -ne 0 ]; then
cd ..
echo -e "${RED}[FAIL]${NONE} install paddledet wheel failed !"
exit 1
fi
echo -e "${BLUE}[install]${NONE} ${GREEN}paddledet install success\n"
cd ..
}
function unittest() {
if [ -d $TEST_DIR ]; then
rm -rf $TEST_DIR
fi;
echo -e "${BLUE}[unittest]${NONE} run unittests..."
# NOTE: perform unittests under TEST_DIR to
# make sure installed paddledet is used
mkdir $TEST_DIR
cp -r $CFG_DIR $TEST_DIR
cp -r $DATA_DIR $TEST_DIR
cd $TEST_DIR
if [ $? != 0 ]; then
exit 1
fi
find "../ppdet" -wholename '*tests/test_*' -type f -print0 | \
xargs -0 -I{} -n1 -t bash -c 'python -u -s {}'
# clean TEST_DIR
cd ..
rm -rf $TEST_DIR
echo -e "${BLUE}[unittest]${NONE} ${GREEN}unittests success\n${NONE}"
}
function cleanup() {
if [ -d $TEST_DIR ]; then
rm -rf $TEST_DIR
fi
rm -rf $BUILD_DIR $EGG_DIR
pip uninstall -y paddledet
}
function abort() {
echo -e "${RED}[FAIL]${NONE} build wheel and unittest failed !
please check your code" 1>&2
cur_dir=`basename "$pwd"`
if [ cur_dir==$TEST_DIR -o cur_dir==$DIST_DIR ]; then
cd ..
fi
rm -rf $BUILD_DIR $EGG_DIR $DIST_DIR $TEST_DIR
pip uninstall -y paddledet
}
python_version_check
trap 'abort' 0
set -e
init
build_and_install
unittest
cleanup
# get Paddle version
PADDLE_VERSION=`python -c "import paddle; print(paddle.version.full_version)"`
PADDLE_COMMIT=`python -c "import paddle; print(paddle.version.commit)"`
PADDLE_COMMIT=`git rev-parse --short $PADDLE_COMMIT`
# get PaddleDetection branch
PPDET_BRANCH=`git rev-parse --abbrev-ref HEAD`
PPDET_COMMIT=`git rev-parse --short HEAD`
# get Python version
PYTHON_VERSION=`python -c "import platform; print(platform.python_version())"`
echo -e "\n${GREEN}paddledet wheel compiled and checked success !${NONE}
${BLUE}Python version:${NONE} $PYTHON_VERSION
${BLUE}Paddle version:${NONE} $PADDLE_VERSION ($PADDLE_COMMIT)
${BLUE}PaddleDetection branch:${NONE} $PPDET_BRANCH ($PPDET_COMMIT)\n"
echo -e "${GREEN}wheel saved under${NONE} ${RED}${BOLD}./dist"
trap : 0
|