caffe+linux平台——跑深度学习的流程

最新推荐文章于 2024-06-25 16:04:26 发布

Miss_yuki

最新推荐文章于 2024-06-25 16:04:26 发布

阅读量1.7k

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/Miss_yuki/article/details/79158345

版权

本文介绍了在Linux环境中使用Caffe进行深度学习的步骤，包括准备图像、创建文件列表、训练模型和测试，并提供了相关脚本代码。适合初学者参考。

摘要由CSDN通过智能技术生成

大家好，放假在即，来整理一波在Linux上用caffe跑深度学习的流程，免得开学回来忘记。

以下地址均为我自己电脑上的地址，大家要照着跑请自行修改地址哦！

1.将训练和测试的图像放入examples-->images.训练图像放入train文件夹中，测试图像放入test文件夹中。

2.在自己创建的examples-->mytest中，创建creat_filelist.sh来生成对应标签train.txt和test.txt，具体代码如下：

#!/usr/bin/env sh
DATA=examples/images/
MY=examples/mytest

echo "Create train.txt..."
rm -rf $MY/train.txt

for i in 0 1 2 3 4 5 6 7 8 9 10
do
find $DATA/train -name $i*.jpg|cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/train.txt
done

echo "Creat test.txt..."
rm -rf $MY/test.txt

for i in 0 1 2 3 4 5 6 7 8 9 10
do
find $DATA/test -name $i*.jpg | cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/test.txt
done
echo "all done.."

3. 在自己创建的examples-->mytest中，创建creat_mytest.sh来将train和test的图片转换成LMD