Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
26b13269
Commit
26b13269
authored
Nov 09, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[FasterRCNN] fix image shape bug..
parent
8543db08
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
examples/FasterRCNN/train.py
examples/FasterRCNN/train.py
+2
-2
No files found.
examples/FasterRCNN/train.py
View file @
26b13269
...
...
@@ -75,17 +75,17 @@ class Model(ModelDesc):
all_anchors
=
tf
.
constant
(
get_all_anchors
(),
name
=
'all_anchors'
,
dtype
=
tf
.
float32
)
fm_anchors
=
tf
.
slice
(
all_anchors
,
[
0
,
0
,
0
,
0
],
tf
.
stack
([
tf
.
shape
(
image
)[
0
]
//
config
.
ANCHOR_STRIDE
,
tf
.
shape
(
image
)[
1
]
//
config
.
ANCHOR_STRIDE
,
tf
.
shape
(
image
)[
2
]
//
config
.
ANCHOR_STRIDE
,
-
1
,
-
1
]),
name
=
'fm_anchors'
)
return
fm_anchors
def
_build_graph
(
self
,
inputs
):
is_training
=
get_current_tower_context
()
.
is_training
image
,
anchor_labels
,
anchor_boxes
,
gt_boxes
,
gt_labels
=
inputs
fm_anchors
=
self
.
_get_anchors
(
image
)
image
=
self
.
_preprocess
(
image
)
fm_anchors
=
self
.
_get_anchors
(
image
)
anchor_boxes_encoded
=
encode_bbox_target
(
anchor_boxes
,
fm_anchors
)
featuremap
=
pretrained_resnet_conv4
(
image
,
config
.
RESNET_NUM_BLOCK
[:
3
])
rpn_label_logits
,
rpn_box_logits
=
rpn_head
(
featuremap
,
1024
,
config
.
NR_ANCHOR
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment