Example #1
0
// Usage: caffe_('net_copy_from', hNet, weights_file)
static void net_copy_from(MEX_ARGS) {
	mxCHECK(nrhs == 2 && mxIsStruct(prhs[0]) && mxIsChar(prhs[1]),
		"Usage: caffe_('net_copy_from', hNet, weights_file)");
	Net<float>* net = handle_to_ptr<Net<float> >(prhs[0]);
	char* weights_file = mxArrayToString(prhs[1]);
	mxCHECK_FILE_EXIST(weights_file);
	net->CopyTrainedLayersFrom(weights_file);
	mxFree(weights_file);
}
Example #2
0
EXPORT void caffe_net_CopyTrainedLayersFrom(void *netAnon, char *trained_filename)
{
	Net<float> *net = (Net<float> *)netAnon;
	net->CopyTrainedLayersFrom(trained_filename);
}