remove masks requirements (intel-analytics#2959)

dding3 · Nov 18, 2019 · 14bcf67 · 14bcf67
1 parent 0c04895
commit 14bcf67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...a/com/intel/analytics/bigdl/dllib/feature/transform/vision/image/label/roi/RoiLabel.scala b/...a/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