Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
weixin_47816946
simple-faster-rcnn-pytorch
提交
214c5553
S
simple-faster-rcnn-pytorch
项目概览
weixin_47816946
/
simple-faster-rcnn-pytorch
与 Fork 源项目一致
从无法访问的项目Fork
通知
8
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
S
simple-faster-rcnn-pytorch
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
214c5553
编写于
12月 21, 2017
作者:
C
chenyuntc
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add support for warm start
上级
50f8eaf9
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
15 addition
and
5 deletion
+15
-5
.gitignore
.gitignore
+1
-0
config.py
config.py
+1
-0
model/faster_rcnn_vgg16.py
model/faster_rcnn_vgg16.py
+8
-3
train.py
train.py
+5
-2
未找到文件。
.gitignore
浏览文件 @
214c5553
...
@@ -14,3 +14,4 @@ model/utils/build/
...
@@ -14,3 +14,4 @@ model/utils/build/
imgs/
imgs/
*.png
*.png
*.jpg
*.jpg
misc/
config.py
浏览文件 @
214c5553
...
@@ -42,6 +42,7 @@ class Config:
...
@@ -42,6 +42,7 @@ class Config:
use_adam
=
False
use_adam
=
False
use_chainer
=
False
use_chainer
=
False
use_drop
=
False
# debug
# debug
debug_file
=
'/tmp/debugf'
debug_file
=
'/tmp/debugf'
...
...
model/faster_rcnn_vgg16.py
浏览文件 @
214c5553
...
@@ -15,9 +15,14 @@ def decom_vgg16(pretrained=True):
...
@@ -15,9 +15,14 @@ def decom_vgg16(pretrained=True):
model
=
vgg16
(
pretrained
)
model
=
vgg16
(
pretrained
)
features
=
list
(
model
.
features
)[:
30
]
features
=
list
(
model
.
features
)[:
30
]
classifier
=
model
.
classifier
classifier
=
model
.
classifier
# classifier = list(classifier)
# del the last layer
classifier
=
list
(
classifier
)
del
classifier
.
_modules
[
'6'
]
# delete dropout
del
classifier
[
6
]
if
not
opt
.
use_drop
:
del
classifier
[
5
]
del
classifier
[
2
]
classifier
=
nn
.
Sequential
(
*
classifier
)
# free top3 conv
# free top3 conv
for
layer
in
features
[:
10
]:
for
layer
in
features
[:
10
]:
...
...
train.py
浏览文件 @
214c5553
...
@@ -62,6 +62,7 @@ def train(**kwargs):
...
@@ -62,6 +62,7 @@ def train(**kwargs):
trainer
.
load
(
opt
.
load_path
)
trainer
.
load
(
opt
.
load_path
)
print
(
'load pretrained model from %s'
%
opt
.
load_path
)
print
(
'load pretrained model from %s'
%
opt
.
load_path
)
trainer
.
optimizer
=
trainer
.
faster_rcnn
.
get_great_optimizer
()
trainer
.
vis
.
text
(
dataset
.
db
.
label_names
,
win
=
'labels'
)
trainer
.
vis
.
text
(
dataset
.
db
.
label_names
,
win
=
'labels'
)
best_map
=
0
best_map
=
0
for
epoch
in
range
(
opt
.
epoch
):
for
epoch
in
range
(
opt
.
epoch
):
...
@@ -98,7 +99,7 @@ def train(**kwargs):
...
@@ -98,7 +99,7 @@ def train(**kwargs):
trainer
.
vis
.
text
(
str
(
trainer
.
rpn_cm
.
value
().
tolist
()),
win
=
'rpn_cm'
)
trainer
.
vis
.
text
(
str
(
trainer
.
rpn_cm
.
value
().
tolist
()),
win
=
'rpn_cm'
)
# roi confusion matrix
# roi confusion matrix
trainer
.
vis
.
img
(
'roi_cm'
,
at
.
totensor
(
trainer
.
roi_cm
.
conf
,
False
).
float
())
trainer
.
vis
.
img
(
'roi_cm'
,
at
.
totensor
(
trainer
.
roi_cm
.
conf
,
False
).
float
())
if
best_map
>
0.6
:
if
best_map
>
0.6
and
opt
.
test_num
<
5000
:
opt
.
test_num
=
10000
opt
.
test_num
=
10000
best_map
=
0
best_map
=
0
eval_result
=
eval
(
test_dataloader
,
faster_rcnn
,
test_num
=
opt
.
test_num
)
eval_result
=
eval
(
test_dataloader
,
faster_rcnn
,
test_num
=
opt
.
test_num
)
...
@@ -106,9 +107,11 @@ def train(**kwargs):
...
@@ -106,9 +107,11 @@ def train(**kwargs):
if
eval_result
[
'map'
]
>
best_map
:
if
eval_result
[
'map'
]
>
best_map
:
best_map
=
eval_result
[
'map'
]
best_map
=
eval_result
[
'map'
]
best_path
=
trainer
.
save
(
best_map
=
best_map
)
best_path
=
trainer
.
save
(
best_map
=
best_map
)
else
:
if
epoch
==
8
:
trainer
.
load
(
best_path
)
trainer
.
load
(
best_path
)
trainer
.
faster_rcnn
.
scale_lr
(
opt
.
lr_decay
)
trainer
.
faster_rcnn
.
scale_lr
(
opt
.
lr_decay
)
if
epoch
==
0
:
trainer
.
optimizer
=
trainer
.
faster_rcnn
.
get_optimizer
()
trainer
.
vis
.
plot
(
'test_map'
,
eval_result
[
'map'
])
trainer
.
vis
.
plot
(
'test_map'
,
eval_result
[
'map'
])
lr_
=
trainer
.
faster_rcnn
.
optimizer
.
param_groups
[
0
][
'lr'
]
lr_
=
trainer
.
faster_rcnn
.
optimizer
.
param_groups
[
0
][
'lr'
]
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录