From bb57abed4eac03d05aac8bfa3e8adfd5368d8db4 Mon Sep 17 00:00:00 2001 From: gigasquid Date: Fri, 9 Nov 2018 15:43:49 -0500 Subject: [PATCH] Fix problem with some OSX not handling the cast on imDecode --- scala-package/core/src/main/scala/org/apache/mxnet/Image.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/Image.scala b/scala-package/core/src/main/scala/org/apache/mxnet/Image.scala index 43f81a22a409..77881ab940be 100644 --- a/scala-package/core/src/main/scala/org/apache/mxnet/Image.scala +++ b/scala-package/core/src/main/scala/org/apache/mxnet/Image.scala @@ -42,7 +42,7 @@ object Image { def imDecode(buf: Array[Byte], flag: Int, to_rgb: Boolean, out: Option[NDArray]): NDArray = { - val nd = NDArray.array(buf.map(_.toFloat), Shape(buf.length)) + val nd = NDArray.array(buf.map( x => (x & 0xFF).toFloat), Shape(buf.length)) val byteND = NDArray.api.cast(nd, "uint8") val args : ListBuffer[Any] = ListBuffer() val map : mutable.Map[String, Any] = mutable.Map()