Skip to content

Commit 925ccaf

Browse files
committed
[fix] installation of yolox
1 parent 84b1d44 commit 925ccaf

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

scripts/install.sh

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ mkdir -p ${MODELS_WEIGHT_DIR}
6969
pip3 install -U pip wheel
7070

7171
## Detectron2
72-
if [ ${MODEL} == "detectron2" ] || [ ${MODEL} == "all" ]; then
72+
if [ "${MODEL,,}" == "detectron2" ] || [ ${MODEL} == "all" ]; then
7373

7474
echo
7575
echo "Installing detectron2"
@@ -143,7 +143,7 @@ if [ ${MODEL} == "detectron2" ] || [ ${MODEL} == "all" ]; then
143143
fi
144144
fi
145145

146-
if [ ${MODEL} == "JDE" ] || [ ${MODEL} == "all" ]; then
146+
if [ "${MODEL,,}" == "jde" ] || [ ${MODEL} == "all" ]; then
147147

148148
echo
149149
echo "Installing JDE"
@@ -220,19 +220,23 @@ if [ ${MODEL} == "JDE" ] || [ ${MODEL} == "all" ]; then
220220
# fi
221221
fi
222222

223-
if [ ${MODEL} == "YOLOX" ] || [ ${MODEL} == "all" ]; then
223+
if [ "${MODEL,,}" == "yolox" ] || [ ${MODEL} == "all" ]; then
224224
echo
225225
echo "Installing YOLOX (reference: https://github.com/Megvii-BaseDetection/YOLOX)"
226226
echo
227227

228228
# clone
229229
if [ -z "$(ls -A ${MODELS_SOURCE_DIR}/yolox)" ]; then
230230
git clone https://github.com/Megvii-BaseDetection/YOLOX.git ${MODELS_SOURCE_DIR}/yolox
231-
# fixed to a specific commit on Nov.19, 2024 for the compatibility in the future.
231+
232+
# checkout specific commit on Nov.19, 2024 for now to avoid compatibility in the future.
233+
cd ${MODELS_SOURCE_DIR}/yolox
232234
git reset --hard d872c71bf63e1906ef7b7bb5a9d7a529c7a59e6a
235+
cd ${SCRIPT_DIR}/..
233236
fi
237+
234238
cd ${MODELS_SOURCE_DIR}/yolox
235-
# miminum requirments - no onnyx, etc.
239+
# miminum requirments - no onnx, etc.
236240
cp ${SCRIPT_DIR}/yolox_requirements.txt requirements.txt
237241

238242
pip3 install -e .

0 commit comments

Comments
 (0)