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

来源:互联网 发布:mysql经典合集 编辑:程序博客网 时间:2024/05/17 01:48
#!/usr/bin/env sh# This script test four voc images using faster rcnn end-to-end trained model (ZF-Model)set -e#从if到fi,其作用是判断选用哪一个gpuif [ ! -n "$1" ] ;then    #如果该shell是带参数的,而且其第一个参数"$1"是非空的,-n是判断一个判断式,后面的字符串非空则返回非零;空则返回0.通过前面的否定表达式!,[ ! -n "$1" ]的意思是,如果不带参数,则gpu=0    echo "\$1 is empty, default is 0"    gpu=0else    echo "use $1-th gpu"    gpu=$1fiCAFFE=build/tools/caffe#其默认的目录是caffe-reid目录,MODEL下有train.proto等文件MODEL=models/market1501/caffenet#下面由GLOG_log_dir开始的一长串是用来指定输出日志的位置的。后面接的是$CAFFE train,这是一个caffe的命令,用于开始模型的训练,该命令后面的均是train的参数。#gpu参数#solve的位置#weights表示已有模型,说明此时的训练是在已有模型上进行finetuningGLOG_log_dir=$MODEL/log $CAFFE train \    --gpu $gpu \    --solver models/market1501/caffenet/solver.proto \    --weights models/pretrain_model/bvlc_reference_caffenet.caffemodel
0 0
原创粉丝点击