lanking520 commented on a change in pull request #12848: Fix Batch input issue
with Scala Benchmark
URL: https://github.com/apache/incubator-mxnet/pull/12848#discussion_r226483031
##########
File path:
scala-package/infer/src/main/scala/org/apache/mxnet/infer/ImageClassifier.scala
##########
@@ -182,7 +180,7 @@ object ImageClassifier {
resizedImage.flush()
// creating NDArray according to the input shape
- val pixelsArray = NDArray.array(result, shape = inputImageShape)
+ val pixelsArray = NDArray.array(result, shape = new Shape(1 +:
inputImageShape.toVector))
Review comment:
@nswamy as we discussed offline, this is just a design decision. As an
alternative, we will request the user to add a reshape if they would like to
concatenate the Image NDArray altogether in one piece.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
