Merge pull request #2009 from heavengate/fix_yolo_downlod

fix weights/download.sh.

PaddleCV/yolov3/README_cn.md

@@ -64,7 +64,7 @@ YOLOv3 的网络结构由基础特征提取网络、multi-scale特征融合层
 
 **下载预训练模型：** 本示例提供darknet53预训练模型，该模型转换自作者提供的darknet53在ImageNet上预训练的权重，采用如下命令下载预训练模型：
 
-    sh ./weights/download_pretrained_weight.sh
+    sh ./weights/download.sh
 
 通过初始化`pretrain` 加载预训练模型。同时在参数微调时也采用该设置加载已训练模型。
 请在训练前确认预训练模型下载与加载正确，否则训练过程中损失可能会出现NAN。

PaddleCV/yolov3/train.py

@@ -74,6 +74,8 @@ def train():
     exe.run(fluid.default_startup_program())
 
     if cfg.pretrain:
+        if not os.path.exists(cfg.pretrain):
+            print("Pretrain weights not found: {}".format(cfg.pretrain))
         def if_exist(var):
             return os.path.exists(os.path.join(cfg.pretrain, var.name))
         fluid.io.load_vars(exe, cfg.pretrain, predicate=if_exist)

PaddleCV/yolov3/weights/download.sh

-DIR="$(dirname "$PWD -P")"
+DIR="$( cd "$(dirname "$0")" ; pwd -P )"
 cd "$DIR"
 
 # Download the pretrain weights.