Skip to content

GitLab

Commit 782d3852 authored by syiming's avatar syiming
Browse files

fix code style

parent 740ef06b
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
View file @ 782d3852
...@@ -1956,7 +1956,8 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -1956,7 +1956,8 @@ class FasterRCNNMetaArch(model.DetectionModel):
box_levels = None box_levels = None
if num_levels != 1: if num_levels != 1:
# If there are mutiple levels to select, get the box levels # If there are mutiple levels to select, get the box levels
box_levels = ops.fpn_feature_levels(num_levels, num_levels - 2, box_levels = ops.fpn_feature_levels(
num_levels, num_levels - 2,
tf.sqrt(self._resize_shape[1] * self._resize_shape[2] * 1.0) / 224.0, tf.sqrt(self._resize_shape[1] * self._resize_shape[2] * 1.0) / 224.0,
proposal_boxes_normalized) proposal_boxes_normalized)
...@@ -2528,7 +2529,6 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -2528,7 +2529,6 @@ class FasterRCNNMetaArch(model.DetectionModel):
box_list.BoxList(tf.reshape(proposal_boxes, [-1, 4])), box_list.BoxList(tf.reshape(proposal_boxes, [-1, 4])),
image_shape[1], image_shape[2], check_range=False).get() image_shape[1], image_shape[2], check_range=False).get()
# TODO(syiming): check this!
flat_cropped_gt_mask = self._crop_and_resize_fn( flat_cropped_gt_mask = self._crop_and_resize_fn(
[tf.expand_dims(flat_gt_masks, -1)], [tf.expand_dims(flat_gt_masks, -1)],
tf.expand_dims(flat_normalized_proposals, axis=1), None, tf.expand_dims(flat_normalized_proposals, axis=1), None,
......
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