Fix and add test for retinanet postprocessing

vanvalenlab · Sep 5, 2019 · e765672 · e765672
1 parent 327dcb5
commit e765672
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/redis_consumer/processing_test.py b/redis_consumer/processing_test.py
@@ -116,7 +116,14 @@ def test_watershed():
 
 
 def test_retinanet():
+    im = _sample1(10, 10, 40, 40)
+    out = _retinanet_data(im)[:-1]
+
+    label = processing.retinanet_to_label_image(out, 40, 40)
+
+
+def test_retinanet_semantic():
     im = _sample1(10, 10, 40, 40)
     out = _retinanet_data(im)
 
-    label = processing.retinanet_to_label_image(out)
+    label = processing.retinanet_semantic_to_label_image(out)