Commit 2b29d716 authored by Pengchong Jin's avatar Pengchong Jin Committed by A. Unique TensorFlower
Browse files

Rename sampling_ops to target_ops.

PiperOrigin-RevId: 308746074
parent 0886b384
...@@ -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)
......
...@@ -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.
# ============================================================================== # ==============================================================================
"""Sampling related ops.""" """Target and sampling related ops."""
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import division from __future__ import division
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment