Commit 599f61a4 authored by Vishnu Banna's avatar Vishnu Banna
Browse files

Merge branch 'kmeans3' of https://github.com/PurdueDualityLab/tf-models into kmeans3

parents bd09680a a43c2e1c
......@@ -274,7 +274,8 @@ class BoxGenInputReader(input_reader.InputReader):
k: `int` for the number for centroids to generate.
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
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
anchor boxes. Must be in the set {across_level, per_level}.
image_resolution: `List[int]` for the resolution of the boxes to run
......@@ -304,4 +305,4 @@ class BoxGenInputReader(input_reader.InputReader):
logging.info('clusting complete -> default boxes used ::')
logging.info(boxes)
return boxes
\ No newline at end of file
return boxes
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