update comments

apache · Oct 20, 2018 · e3e8d19 · e3e8d19
1 parent c4c98ef
commit e3e8d19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scala-package/infer/src/main/scala/org/apache/mxnet/infer/ImageClassifier.scala b/scala-package/infer/src/main/scala/org/apache/mxnet/infer/ImageClassifier.scala
@@ -154,7 +154,7 @@ object ImageClassifier {
     * @param resizedImage     BufferedImage to get pixels from
     * @param inputImageShape  Input shape; for example for resnet it is (3,224,224).
                               Should be same as inputDescriptor shape.
-    * @return                 NDArray pixels array with shape (1, 3, 224, 224) in NCHW format
+    * @return                 NDArray pixels array with shape (3, 224, 224) in CHW format
     */
   def bufferedImageToPixels(resizedImage: BufferedImage, inputImageShape: Shape): NDArray = {
     // Get height and width of the image