<span style="font-family:SimHei;font-size:24px;">%%
clc; close all; clear all;
use_gpu=1; % GPU 模式
if exist('../+caffe', 'dir')
addpath('..');
else
error('Please run this demo from caffe/matlab/demo');
end
if exist('use_gpu', 'var') && use_gpu
caffe.set_mode_gpu();
gpu_id = 0; % we will use the first gpu in this demo
caffe.set_device(gpu_id);
else
caffe.set_mode_cpu();
end
model_dir = '../../models/bvlc_alexnet/';
net_model = [model_dir 'train_val.prototxt'];
% net_model = [model_dir '__deploy.prototxt'];
net_weights = [model_dir 'caffe_alexnet_train_iter_310000.caffemodel'];
% net_weights = [model_dir 'bvlc_alexnet.caffemodel'];
phase = 'train';
if ~exist(net_weights, 'file')
error('Please download CaffeNet from Model Zoo before you run this demo'
HAT_dataset compute accuracy for caffe_deep learning approach of matlab interface--->>> code record
最新推荐文章于 2024-07-25 20:53:45 发布