Unverified Commit a43c2e1c authored by Vishnu Banna's avatar Vishnu Banna Committed by GitHub
Browse files

documentation update for scaling mode

parent f59c0d29
...@@ -274,7 +274,8 @@ class BoxGenInputReader(input_reader.InputReader): ...@@ -274,7 +274,8 @@ class BoxGenInputReader(input_reader.InputReader):
k: `int` for the number for centroids to generate. k: `int` for the number for centroids to generate.
anchors_per_scale: `int` for how many anchor boxes to use per level. anchors_per_scale: `int` for how many anchor boxes to use per level.
scaling_mode: `str` for the type of box scaling to used when generating scaling_mode: `str` for the type of box scaling to used when generating
anchor boxes. Must be in the set {sqrt_log, default}. anchor boxes. Must be in the set {sqrt, none}. By default we use sqrt
to get an even distribution of anchor boxes across FPN levels.
box_generation_mode: `str` for the type of kmeans to use when generating box_generation_mode: `str` for the type of kmeans to use when generating
anchor boxes. Must be in the set {across_level, per_level}. anchor boxes. Must be in the set {across_level, per_level}.
image_resolution: `List[int]` for the resolution of the boxes to run image_resolution: `List[int]` for the resolution of the boxes to run
...@@ -304,4 +305,4 @@ class BoxGenInputReader(input_reader.InputReader): ...@@ -304,4 +305,4 @@ class BoxGenInputReader(input_reader.InputReader):
logging.info('clusting complete -> default boxes used ::') logging.info('clusting complete -> default boxes used ::')
logging.info(boxes) logging.info(boxes)
return boxes return boxes
\ No newline at end of file
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