2121
2222
2323class COCORecallTest (tf .test .TestCase ):
24-
2524 def test_runs_inside_model (self ):
26- i = keras .layers .Input ((None , None , 6 ))
25+ i = keras .layers .Input ((None , None , 6 ))
2726 model = keras .Model (i , i )
2827
2928 recall = COCORecall (
@@ -33,7 +32,9 @@ def test_runs_inside_model(self):
3332 )
3433
3534 # These would match if they were in the area range
36- y_true = np .array ([[[0 , 0 , 10 , 10 , 1 ], [5 , 5 , 10 , 10 , 1 ]]]).astype (np .float32 )
35+ y_true = np .array ([[[0 , 0 , 10 , 10 , 1 ], [5 , 5 , 10 , 10 , 1 ]]]).astype (
36+ np .float32
37+ )
3738 y_pred = np .array ([[[0 , 0 , 10 , 10 , 1 , 1.0 ], [5 , 5 , 10 , 10 , 1 , 0.9 ]]]).astype (
3839 np .float32
3940 )
@@ -51,7 +52,9 @@ def test_recall_area_range_filtering(self):
5152 )
5253
5354 # These would match if they were in the area range
54- y_true = np .array ([[[0 , 0 , 10 , 10 , 1 ], [5 , 5 , 10 , 10 , 1 ]]]).astype (np .float32 )
55+ y_true = np .array ([[[0 , 0 , 10 , 10 , 1 ], [5 , 5 , 10 , 10 , 1 ]]]).astype (
56+ np .float32
57+ )
5558 y_pred = np .array ([[[0 , 0 , 10 , 10 , 1 , 1.0 ], [5 , 5 , 10 , 10 , 1 , 0.9 ]]]).astype (
5659 np .float32
5760 )
0 commit comments