Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
N
ngraph
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ngraph
Commits
2aa508b4
Commit
2aa508b4
authored
Aug 02, 2019
by
Mateusz Bencer
Committed by
Scott Cyphers
Aug 02, 2019
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Adjust op ROIPooling (#3346)
* Adjust ROIPolling * Fixed comment
parent
0850d234
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
8 deletions
+8
-8
roi_pooling.cpp
src/ngraph/op/experimental/layers/roi_pooling.cpp
+4
-4
roi_pooling.hpp
src/ngraph/op/experimental/layers/roi_pooling.hpp
+4
-4
No files found.
src/ngraph/op/experimental/layers/roi_pooling.cpp
View file @
2aa508b4
...
...
@@ -22,14 +22,14 @@ using namespace ngraph;
const
string
op
::
ROIPooling
::
type_name
{
"ROIPooling"
};
op
::
ROIPooling
::
ROIPooling
(
const
shared_ptr
<
Node
>&
input
,
const
s
td
::
s
hared_ptr
<
Node
>&
coords
,
const
shared_ptr
<
Node
>&
coords
,
const
Shape
&
output_size
,
const
float
spatial_scale
,
const
st
d
::
string
&
kin
d
)
const
st
ring
&
metho
d
)
:
Op
(
check_single_output_args
({
input
,
coords
}))
,
m_output_size
(
output_size
)
,
m_spatial_scale
(
spatial_scale
)
,
m_
kind
(
kin
d
)
,
m_
method
(
metho
d
)
{
constructor_validate_and_infer_types
();
}
...
...
@@ -69,5 +69,5 @@ shared_ptr<Node> op::ROIPooling::copy_with_new_args(const NodeVector& new_args)
{
check_new_args_count
(
this
,
new_args
);
return
make_shared
<
ROIPooling
>
(
new_args
.
at
(
0
),
new_args
.
at
(
1
),
m_output_size
,
m_spatial_scale
,
m_
kin
d
);
new_args
.
at
(
0
),
new_args
.
at
(
1
),
m_output_size
,
m_spatial_scale
,
m_
metho
d
);
}
src/ngraph/op/experimental/layers/roi_pooling.hpp
View file @
2aa508b4
...
...
@@ -34,12 +34,12 @@ namespace ngraph
/// \param coords Coordinates of bounding boxes
/// \param output_size Height/Width of ROI output features
/// \param spatial_scale Ratio of input feature map over input image size
/// \param
kind Kin
d of pooling - Max or Bilinear
/// \param
method Metho
d of pooling - Max or Bilinear
ROIPooling
(
const
std
::
shared_ptr
<
Node
>&
input
,
const
std
::
shared_ptr
<
Node
>&
coords
,
const
Shape
&
output_size
,
const
float
spatial_scale
,
const
std
::
string
&
kin
d
);
const
std
::
string
&
metho
d
);
void
validate_and_infer_types
()
override
;
...
...
@@ -48,11 +48,11 @@ namespace ngraph
const
Shape
&
get_output_size
()
const
{
return
m_output_size
;
}
float
get_spatial_scale
()
const
{
return
m_spatial_scale
;
}
const
std
::
string
&
get_
kind
()
const
{
return
m_kin
d
;
}
const
std
::
string
&
get_
method
()
const
{
return
m_metho
d
;
}
private
:
Shape
m_output_size
;
float
m_spatial_scale
;
std
::
string
m_
kin
d
;
std
::
string
m_
metho
d
;
};
}
}
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