提交 181a0d33 编写于 作者: 彭冲 提交者: Waleed

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

上级 85d574c5
......@@ -2198,6 +2198,7 @@ class MaskRCNN():
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)))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册