From 14bcf676b62619f4d33c9ce71963fca22759f491 Mon Sep 17 00:00:00 2001 From: zhangxiaoli73 <380761639@qq.com> Date: Mon, 18 Nov 2019 22:02:38 +0800 Subject: [PATCH] remove masks requirements (#2959) --- .../feature/transform/vision/image/label/roi/RoiLabel.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/feature/transform/vision/image/label/roi/RoiLabel.scala b/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/feature/transform/vision/image/label/roi/RoiLabel.scala index cb038780450..ae57a49b0a1 100644 --- a/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/feature/transform/vision/image/label/roi/RoiLabel.scala +++ b/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/feature/transform/vision/image/label/roi/RoiLabel.scala @@ -57,7 +57,7 @@ case class RoiLabel(classes: Tensor[Float], bboxes: Tensor[Float], def toTable: Table = { val table = T() if (masks != null) { - require(masks.length > 0, "The masks can either be null or a non-empty array") + // masks may be empty array table(RoiImageInfo.MASKS) = masks.map(_.toRLE) } table(RoiImageInfo.CLASSES) = classes