in detect function,length of images must be equal to BATCH_SIZE

kezpitt · Nov 10, 2017 · 181a0d3 · 181a0d3
1 parent 85d574c
commit 181a0d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model.py b/model.py
@@ -2198,6 +2198,7 @@ def detect(self, images, verbose=0):
         masks: [H, W, N] instance binary masks
         """
         assert self.mode == "inference", "Create model in inference mode."
+        assert len(images) == self.config.BATCH_SIZE, "len(images) must be equal to BATCH_SIZE"
 
         if verbose: 
             log("Processing {} images".format(len(images)))