Номер метки TensorFlow не соответствует форме на оси - PullRequest
Новая
       8

Номер метки TensorFlow не соответствует форме на оси

0 голосов
/ 01 февраля 2020

пытается запустить кодовую метку: https://codelabs.developers.google.com/codelabs/recognize-flowers-with-tensorflow-on-android/#6

Я разработал свои собственные файлы и файлы списков, хотя, похоже, у меня появляется эта ошибка, когда я запускаю ее через Android :

java.lang.IllegalArgumentException: Label number 6 mismatch the shape on axis 1

Затем он связывает меня с этой частью кода: 

Map<String, Float> labeledProbability =
        new TensorLabel(labels, probabilityProcessor.process(outputProbabilityBuffer))
            .getMapWithFloatValue();
    Trace.endSection();

это в разделе результатов классификации 

public List<Recognition> recognizeImage(final Bitmap bitmap, int sensorOrientation) {
    // Logs this method so that it can be analyzed with systrace.
    Trace.beginSection("recognizeImage");

    Trace.beginSection("loadImage");
    long startTimeForLoadImage = SystemClock.uptimeMillis();
    inputImageBuffer = loadImage(bitmap, sensorOrientation);
    long endTimeForLoadImage = SystemClock.uptimeMillis();
    Trace.endSection();
    LOGGER.v("Timecost to load the image: " + (endTimeForLoadImage - startTimeForLoadImage));

    // Runs the inference call.
    Trace.beginSection("runInference");
    long startTimeForReference = SystemClock.uptimeMillis();
    tflite.run(inputImageBuffer.getBuffer(), outputProbabilityBuffer.getBuffer().rewind());
    long endTimeForReference = SystemClock.uptimeMillis();
    Trace.endSection();
    LOGGER.v("Timecost to run model inference: " + (endTimeForReference - startTimeForReference));

    // Gets the map of label and probability.
    Map<String, Float> labeledProbability =
        new TensorLabel(labels, probabilityProcessor.process(outputProbabilityBuffer))
            .getMapWithFloatValue();
    Trace.endSection();

    // Gets top-k results.
    return getTopKProbability(labeledProbability);
  }

Понятия не имею, почему это работает, кто-нибудь может помочь ??

...