一、数据集合的收集
从网络上下载了五种物体的图片包括:bus, drogon, elep, flow, horse。每类图片120张,建立三个文件夹train,test,val分别存放这5类图片,train每类存放80张,test和val每类都是20张。结构如下:
二、在digist中创建lmdb数据集合
选择界面如下所示:
选择完后点击creat开始lmdb数据集的创建
三、模型文件的配置
搭建自己的网络进行模型的训练,点击custom network按钮,在文字框里输入自己需要搭建的网络的配置信息,去掉训练数据和验证数据及其均值(因为我们选择的是digist创建的lmdb数据集,它会将数据源直接同配置文件进行连接),并将网络的输出层改为5(因为我们只有5类)。
注意:di
Ubuntu系统,用digist可视化界面实现自己的数据训练和测试
最新推荐文章于 2024-05-07 15:30:13 发布