Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Ayu___
fast-rcnn
提交
1a563926
F
fast-rcnn
项目概览
Ayu___
/
fast-rcnn
与 Fork 源项目一致
从无法访问的项目Fork
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
F
fast-rcnn
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
1a563926
编写于
5月 04, 2015
作者:
R
Ross Girshick
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
rm accidentally added file
上级
01ee5ba2
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
0 addition
and
63 deletion
+0
-63
matlab/init_cnn_model.m
matlab/init_cnn_model.m
+0
-63
未找到文件。
matlab/init_cnn_model.m
已删除
100644 → 0
浏览文件 @
01ee5ba2
function
cnn
=
init_cnn_model
(
varargin
)
% cnn = init_cnn_model
% Initialize a CNN with caffe
%
% Optional arguments
% net_file network binary file
% def_file network prototxt file
% use_gpu set to false to use CPU (default: true)
% use_caffe set to false to avoid using caffe (default: true)
% useful for running on the cluster (must use cached pyramids!)
% ------------------------------------------------------------------------
% Options
ip
=
inputParser
;
% network binary file
ip
.
addParamValue
(
'net_file'
,
...
'./data/caffe_nets/ilsvrc_2012_train_iter_310k'
,
...
@
isstr
);
% network prototxt file
ip
.
addParamValue
(
'def_file'
,
...
'./model-defs/pyramid_cnn_output_conv5_scales_7_plane_1713.prototxt'
,
...
@
isstr
);
% Set use_gpu to false to use the CPU
ip
.
addParamValue
(
'use_gpu'
,
true
,
@
islogical
);
% Set use_caffe to false to avoid using caffe
% (must be used in conjunction with cached features!)
ip
.
addParamValue
(
'use_caffe'
,
true
,
@
islogical
);
ip
.
parse
(
varargin
{:});
opts
=
ip
.
Results
;
% ------------------------------------------------------------------------
cnn
.
binary_file
=
opts
.
net_file
;
cnn
.
definition_file
=
opts
.
def_file
;
cnn
.
init_key
=
-
1
;
% load the ilsvrc image mean
data_mean_file
=
'ilsvrc_2012_mean.mat'
;
assert
(
exist
(
data_mean_file
,
'file'
)
~=
0
);
% input size business isn't likley necessary, but we're doing it
% to be consistent with previous experiments
ld
=
load
(
data_mean_file
);
mu
=
ld
.
image_mean
;
clear
ld
;
input_size
=
227
;
off
=
floor
((
size
(
mu
,
1
)
-
input_size
)/
2
)
+
1
;
%mu = mu(off:off+input_size-1, off:off+input_size-1, :);
%mu = sum(sum(mu, 1), 2) / size(mu, 1) / size(mu, 2);
cnn
.
mu
=
reshape
([
102.9801
,
115.9465
,
122.7717
],
[
1
1
3
]);
if
opts
.
use_caffe
cnn
.
init_key
=
...
caffe
(
'init'
,
cnn
.
definition_file
,
cnn
.
binary_file
);
caffe
(
'set_phase_test'
);
if
opts
.
use_gpu
caffe
(
'set_mode_gpu'
);
else
caffe
(
'set_mode_cpu'
);
end
end
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录