fix script

Author: fateshelled

weights/onnx/download.bash

@@ -1,23 +1,27 @@
 #!/bin/bash
 
-# if $1 is empty
-if [ -z "$1" ]; then
-    echo "Usage: $0 <target-model>"
-    echo "Target-Models :"
-    echo "yolox_tiny, yolox_nano, yolox_s, yolox_m, yolox_l, all"
-    exit 1
-fi
-MODEL=$1
-MODEL_VERSION=0.1.1rc0
-SCRIPT_DIR=$(cd $(dirname $0); pwd)
+function download {
+    # if $1 is empty
+    if [ -z "$1" ]; then
+        echo "Usage: $0 <target-model>"
+        echo "Target-Models :"
+        echo "yolox_tiny, yolox_nano, yolox_s, yolox_m, yolox_l, all"
+        return
+    fi
+    MODEL=$1
+    MODEL_VERSION=0.1.1rc0
+    SCRIPT_DIR=$(cd $(dirname ${BASH_SOURCE:-$0}); pwd)
 
-echo $MODEL
-if [ "$MODEL" = "all" ]; then
-    wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_tiny.onnx -P $SCRIPT_DIR
-    wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_nano.onnx -P $SCRIPT_DIR
-    wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_s.onnx -P $SCRIPT_DIR
-    wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_m.onnx -P $SCRIPT_DIR
-    wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_l.onnx -P $SCRIPT_DIR
-else
-    wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/$MODEL.onnx -P $SCRIPT_DIR
-fi
+    echo $MODEL
+    if [ "$MODEL" = "all" ]; then
+        wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_tiny.onnx -P $SCRIPT_DIR
+        wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_nano.onnx -P $SCRIPT_DIR
+        wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_s.onnx -P $SCRIPT_DIR
+        wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_m.onnx -P $SCRIPT_DIR
+        wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/yolox_l.onnx -P $SCRIPT_DIR
+    else
+        wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/$MODEL_VERSION/$MODEL.onnx -P $SCRIPT_DIR
+    fi
+}
+
+download $1

weights/tensorrt/convert.bash

@@ -9,7 +9,7 @@ function convert {
     fi
 
     MODEL=$1
-    SCRIPT_DIR=$(cd $(dirname $0); pwd)
+    SCRIPT_DIR=$(cd $(dirname ${BASH_SOURCE:-$0}); pwd)
 
     echo "Model Name: ${MODEL}"
     echo ""