Fix dimension order in the image shape

Kotlin · Apr 24, 2023 · 4d43480 · 4d43480
1 parent a8b578f
commit 4d43480
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/impl/src/jvmMain/kotlin/org/jetbrains/kotlinx/dl/impl/preprocessing/image/ImageUtil.kt b/impl/src/jvmMain/kotlin/org/jetbrains/kotlinx/dl/impl/preprocessing/image/ImageUtil.kt
@@ -25,5 +25,5 @@ internal fun BufferedImage.copy(): BufferedImage {
 }
 
 internal fun BufferedImage.getShape(): TensorShape {
-    return TensorShape(width.toLong(), height.toLong(), colorModel.numComponents.toLong())
+    return TensorShape(height.toLong(), width.toLong(), colorModel.numComponents.toLong())
 }