Skip to content

GitLab

Commit 9497f59c authored by Vishnu Banna's avatar Vishnu Banna
Browse files

kmeans

parent e0b33491
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/projects/yolo/ops/kmeans_test.py
View file @ 9497f59c
...@@ -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.
"""box_ops tests.""" """kmeans_test tests."""
from absl.testing import parameterized from absl.testing import parameterized
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
...@@ -20,7 +20,7 @@ import tensorflow as tf ...@@ -20,7 +20,7 @@ import tensorflow as tf
from official.vision.beta.projects.yolo.ops import kmeans_anchors from official.vision.beta.projects.yolo.ops import kmeans_anchors
class InputUtilsTest(parameterized.TestCase, tf.test.TestCase): class KMeansTest(parameterized.TestCase, tf.test.TestCase):
@parameterized.parameters((9, 3, 100)) @parameterized.parameters((9, 3, 100))
def test_kmeans(self, k, anchors_per_scale, samples): def test_kmeans(self, k, anchors_per_scale, samples):
......
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