Skip to content

Commit

Permalink
add nms serialization test (keras-team#2285)
Browse files Browse the repository at this point in the history
* fix nms serialization

* remove the default function

---------

Co-authored-by: Divyashree Sreepathihalli <divyashreepathihalli>
  • Loading branch information
divyashreepathihalli authored and yuvraj-wale committed Feb 8, 2024
1 parent a495ad0 commit 6698fcf
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions keras_cv/layers/serialization_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -377,6 +377,14 @@ class SerializationTest(TestCase):
"from_logits": True,
},
),
(
"NonMaxSuppression",
cv_layers.NonMaxSuppression,
{
"bounding_box_format": "yxyx",
"from_logits": True,
},
),
)
def test_layer_serialization(self, layer_cls, init_args):
# TODO: Some layers are not yet compatible with Keras 3.
Expand Down

0 comments on commit 6698fcf

Please sign in to comment.