代码笔记:caffe-reid中caffe_train.sh的解释

#!/usr/bin/env sh
# This script test four voc images using faster rcnn end-to-end trained model (ZF-Model)
set -e
#从if到fi,其作用是判断选用哪一个gpu
if [ ! -n "$1" ] ;then
    #如果该shell是带参数的,而且其第一个参数"$1"是非空的,-n是判断一个判断式,后面的字符串非空则返回非零;空则返回0.通过前面的否定表达式!,[ ! -n "$1" ]的意思是,如果不带参数,则gpu=0
    echo "\$1 is empty, default is 0"
    gpu=0
else
    echo "use $1-th gpu"
    gpu=$1
fi


CAFFE=build/tools/caffe
#其默认的目录是caffe-reid目录,MODEL下有train.proto等文件
MODEL=models/market1501/caffenet

#下面由GLOG_log_dir开始的一长串是用来指定输出日志的位置的。后面接的是$CAFFE train,这是一个caffe的命令,用于开始模型的训练,该命令后面的均是train的参数。
#gpu参数
#solve的位置
#weights表示已有模型,说明此时的训练是在已有模型上进行finetuning
GLOG_log_dir=$MODEL/log $CAFFE train \
    --gpu $gpu \
    --solver models/market1501/caffenet/solver.proto \
    --weights models/pretrain_model/bvlc_reference_caffenet.caffemodel
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值