Github user viirya commented on a diff in the pull request:
https://github.com/apache/spark/pull/20389#discussion_r163756581
--- Diff: mllib/src/main/scala/org/apache/spark/ml/image/ImageSchema.scala
---
@@ -169,8 +169,7 @@ object ImageSchema {
var offset = 0
for (h <- 0 until height) {
for (w <- 0 until width) {
- val color = new Color(img.getRGB(w, h))
-
+ val color = new Color(img.getRGB(w, h), nChannels == 4)
--- End diff --
Why don't directly use `hasAlpha`?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
