Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
2b29d716
Commit
2b29d716
authored
Apr 27, 2020
by
Pengchong Jin
Committed by
A. Unique TensorFlower
Apr 27, 2020
Browse files
Rename sampling_ops to target_ops.
PiperOrigin-RevId: 308746074
parent
0886b384
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
official/vision/detection/modeling/maskrcnn_model.py
official/vision/detection/modeling/maskrcnn_model.py
+3
-3
official/vision/detection/ops/target_ops.py
official/vision/detection/ops/target_ops.py
+1
-1
No files found.
official/vision/detection/modeling/maskrcnn_model.py
View file @
2b29d716
...
@@ -29,8 +29,8 @@ from official.vision.detection.modeling import losses
...
@@ -29,8 +29,8 @@ from official.vision.detection.modeling import losses
from
official.vision.detection.modeling.architecture
import
factory
from
official.vision.detection.modeling.architecture
import
factory
from
official.vision.detection.ops
import
postprocess_ops
from
official.vision.detection.ops
import
postprocess_ops
from
official.vision.detection.ops
import
roi_ops
from
official.vision.detection.ops
import
roi_ops
from
official.vision.detection.ops
import
sampling_ops
from
official.vision.detection.ops
import
spatial_transform_ops
from
official.vision.detection.ops
import
spatial_transform_ops
from
official.vision.detection.ops
import
target_ops
from
official.vision.detection.utils
import
box_utils
from
official.vision.detection.utils
import
box_utils
...
@@ -51,8 +51,8 @@ class MaskrcnnModel(base_model.Model):
...
@@ -51,8 +51,8 @@ class MaskrcnnModel(base_model.Model):
self
.
_fpn_fn
=
factory
.
multilevel_features_generator
(
params
)
self
.
_fpn_fn
=
factory
.
multilevel_features_generator
(
params
)
self
.
_rpn_head_fn
=
factory
.
rpn_head_generator
(
params
)
self
.
_rpn_head_fn
=
factory
.
rpn_head_generator
(
params
)
self
.
_generate_rois_fn
=
roi_ops
.
ROIGenerator
(
params
.
roi_proposal
)
self
.
_generate_rois_fn
=
roi_ops
.
ROIGenerator
(
params
.
roi_proposal
)
self
.
_sample_rois_fn
=
sampling
_ops
.
ROISampler
(
params
.
roi_sampling
)
self
.
_sample_rois_fn
=
target
_ops
.
ROISampler
(
params
.
roi_sampling
)
self
.
_sample_masks_fn
=
sampling
_ops
.
MaskSampler
(
self
.
_sample_masks_fn
=
target
_ops
.
MaskSampler
(
params
.
architecture
.
mask_target_size
,
params
.
architecture
.
mask_target_size
,
params
.
mask_sampling
.
num_mask_samples_per_image
)
params
.
mask_sampling
.
num_mask_samples_per_image
)
...
...
official/vision/detection/ops/
sampling
_ops.py
→
official/vision/detection/ops/
target
_ops.py
View file @
2b29d716
...
@@ -12,7 +12,7 @@
...
@@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# See the License for the specific language governing permissions and
# limitations under the License.
# limitations under the License.
# ==============================================================================
# ==============================================================================
"""
S
ampling related ops."""
"""
Target and s
ampling related ops."""
from
__future__
import
absolute_import
from
__future__
import
absolute_import
from
__future__
import
division
from
__future__
import
division
...
...
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