Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "8aa6eb10be4cb72ac2e36bccadfcea5f356aea54"
Commit 800a4c00 authored by syiming's avatar syiming
Browse files

add note for the choice of num_levels - 2

parent d03f4fb6
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
View file @ 800a4c00
...@@ -1954,6 +1954,9 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -1954,6 +1954,9 @@ 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
# unit_scale_index: num_levels-2 is chosen based on section 4.2 of
# https://arxiv.org/pdf/1612.03144.pdf and works best for Resnet based
# feature extractor.
box_levels = ops.fpn_feature_levels( box_levels = ops.fpn_feature_levels(
num_levels, num_levels - 2, 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,
......
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