设置GPU的ID号
bool Faster_rcnn::init()
{
//init
//Caffe::SetDevice(0);
Caffe::set_mode(Caffe::CPU);
loadnet();
return true;
}
转载请注明: http://blog.csdn.net/forest_world
设置GPU的ID号
bool Faster_rcnn::init()
{
//init
//Caffe::SetDevice(0);
Caffe::set_mode(Caffe::CPU);
loadnet();
return true;
}