一、准备数据集 点击此处返回总目录 二、制作标签 三、将图片转换为LMDB格式 四、修改网络结构文件 五、修改超参数文件 六、训练模型 七、测试模型
一、准备数据集 1. 根据下面5个链接可以分别下到物种类别的图片。
animal类 下载地址: http://www.robots.ox.ac.uk/~vgg/data/pets/
flowers类 下载地址:http://www.robots.ox.ac.uk/~vgg/data/flowers/
plane类 下载地址:http://www.robots.ox.ac.uk/~vgg/data/airplanes_side/airplanes_side.tar
house类 http://www.robots.ox.ac.uk/~vgg/data/houses/houses.tar
guitar类 http://www.robots.ox.ac.uk/~vgg/data/guitars/guitars.tar
2. 下载完之后,解压:
里面全是一些jpg的图片。 flowers //有1360张 plane //有1074张 guitar //有1030张 house //有1000张 image(animal) //有7393张
3. 在caffe-windows\models下,新建my_models_recognition目录,用来存放我们自己的模型。 在my_models_recognition中如下创建文件夹:
|---data //存放训练数据和测试数据 | |---train | | |---animal | | |---flower | | |---plane | | |---house | | |---guitar | | | |---test | |---animal | |---flower | |---plane | |---house | |---guitar | |---labels //存放标签 | | |---bat //存放脚本 | | |---lmdb //存放转换后的lmdb数据 | | |---model & |
使用自己设计的网络训练自己的图像识别模型
最新推荐文章于 2025-05-13 20:13:31 发布