Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Ayu___
fast-rcnn
提交
ef2b2255
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 搜索 >>
提交
ef2b2255
编写于
3月 05, 2015
作者:
R
Ross Girshick
浏览文件
操作
浏览文件
下载
差异文件
Merge remote-tracking branch 'flip/master'
上级
a7722edd
129d1a70
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
26 addition
and
1 deletion
+26
-1
datasets/imdb.py
datasets/imdb.py
+20
-0
datasets/pascal_voc.py
datasets/pascal_voc.py
+1
-1
fast_rcnn_train.py
fast_rcnn_train.py
+3
-0
finetuning.py
finetuning.py
+2
-0
未找到文件。
datasets/imdb.py
浏览文件 @
ef2b2255
import
os
import
PIL
class
imdb
(
object
):
def
__init__
(
self
,
name
):
...
...
@@ -63,3 +64,22 @@ class imdb(object):
all_boxes[class][image] = [] or np.array of shape #dets x 5
"""
raise
NotImplementedError
def
append_flipped_roidb
(
self
):
num_images
=
len
(
self
.
image_index
)
widths
=
[
PIL
.
Image
.
open
(
self
.
image_path_at
(
i
)).
size
[
0
]
for
i
in
xrange
(
num_images
)]
for
i
in
xrange
(
num_images
):
self
.
roidb
[
i
][
'flipped'
]
=
False
boxes
=
self
.
roidb
[
i
][
'boxes'
].
copy
()
oldx1
=
boxes
[:,
0
].
copy
()
oldx2
=
boxes
[:,
2
].
copy
()
boxes
[:,
0
]
=
widths
[
i
]
-
oldx2
-
1
boxes
[:,
2
]
=
widths
[
i
]
-
oldx1
-
1
assert
(
boxes
[:,
2
]
>=
boxes
[:,
0
]).
all
()
entry
=
{
'boxes'
:
boxes
,
'gt_overlaps'
:
self
.
roidb
[
i
][
'gt_overlaps'
],
'gt_classes'
:
self
.
roidb
[
i
][
'gt_classes'
],
'flipped'
:
True
}
self
.
roidb
.
append
(
entry
)
self
.
_image_index
=
self
.
_image_index
*
2
datasets/pascal_voc.py
浏览文件 @
ef2b2255
...
...
@@ -58,7 +58,7 @@ class pascal_voc(datasets.imdb):
self
.
_image_set
+
'.txt'
)
assert
os
.
path
.
exists
(
image_set_file
)
with
open
(
image_set_file
)
as
f
:
image_index
=
tuple
([
x
.
strip
()
for
x
in
f
.
readlines
()])
image_index
=
[
x
.
strip
()
for
x
in
f
.
readlines
()]
return
image_index
def
_get_default_path
(
self
):
...
...
fast_rcnn_train.py
浏览文件 @
ef2b2255
...
...
@@ -156,6 +156,9 @@ if __name__ == '__main__':
imdb_train
=
datasets
.
pascal_voc
(
'trainval'
,
'2007'
)
# enhance roidb to contain flipped examples
imdb_train
.
append_flipped_roidb
()
# enhance roidb to contain some useful derived quanties
roidb_train
=
training_roidb
(
imdb_train
)
...
...
finetuning.py
浏览文件 @
ef2b2255
...
...
@@ -129,6 +129,8 @@ def _get_image_blob(roidb, scale_inds):
im_scale_factors
=
[]
for
i
in
xrange
(
num_images
):
im
=
cv2
.
imread
(
roidb
[
i
][
'image'
])
if
roidb
[
i
][
'flipped'
]:
im
=
im
[:,
::
-
1
,
:]
im
=
im
.
astype
(
np
.
float32
,
copy
=
False
)
im
-=
conf
.
PIXEL_MEANS
im_shape
=
im
.
shape
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录