如何使用caffe训练vgg模型实现自己的数据集分类
发布网友
发布时间:2022-04-23 22:02
我来回答
共1个回答
热心网友
时间:2023-11-02 10:12
我先问你个问题,你有什么硬件平台?显卡是什么配置?有多大的数据集?
vgg网络不是特别复杂,你可以直接从网上下载到他们的网络和相应参数。但是普通人的电脑估计是训练不出这么好的网络的,主要原因是慢。
就拿2012的image net的数据来说,训练图片大约138G,用GTX1080Ti跑的话,不跑个一两周估计效果都不理想。