《表2 Fine-tuning训练参数设置》
训练开始前,首先生成训练集和测试集文件路径,并添加相应类别标签:躺卧(0)、爬跨(1)、站立/游走(2)。由于数据库和均值文件是在Caffe下进行训练的必要文件,因此需要利用Caffe插件convert_imageset将测试集和训练集分别转换为Caffe可识别的数据库,并利用compute_image_mean插件计算训练集的图像均值,以方便训练。本次训练选择的预加载模型为BVLCAlexNet。训练开始时,结合训练平台的硬件性能和预试验时确定的优选参数,调整AlexNet配置文件中train_val.prototxt和solver.prototxt的相关训练参数设置,调整后的训练参数如表2所示。
图表编号 | XD00135851800 严禁用于非法目的 |
---|---|
绘制时间 | 2020.04.25 |
作者 | 王少华、何东健、刘冬 |
绘制单位 | 西北农林科技大学机械与电子工程学院、农业农村部农业物联网重点实验室、西北农林科技大学机械与电子工程学院、农业农村部农业物联网重点实验室、西北农林科技大学机械与电子工程学院、农业农村部农业物联网重点实验室 |
更多格式 | 高清、无水印(增值服务) |