笔记:caffe ssd gpu训练自己的数据集

养鸿运
2023-12-01

一、caffe环境搭建

环境搭建过程网上有很多教程,可能要踩些坑吧,正常,这里不多赘述,主要是我当时也没做记录,实在难以复现那个过程了。

caffe用的weiliu89,先克隆下来,然后再把ssd分支checkout,代码如下:

git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd

然后是修改caffe的Makefile.config等文件,可以参考网上的教程,我自己的Makefile.config:

## Refer to http://caffe.berkeleyvision.org/installation.html
 类似资料: