Блог

Главная - Вопросы по программированию - Ошибка остановки при использовании next () для кода оценки глубины монокуляра

Ошибка остановки при использовании next () для кода оценки глубины монокуляра

#python #stopiteration

#python #остановка

Вопрос:

Я ссылаюсь https://keras.io/examples/vision/depth_estimation/#visualizing-samples код для одного из моих проектов, однако, я получаю ошибку, связанную с StopIteration. Не мог бы кто-нибудь подсказать, как это можно исправить?

 def visualize_depth_map(samples, test=False, model=None):
    input, target = samples
    cmap = plt.cm.jet
    cmap.set_bad(color="black")

    if test:
        pred = model.predict(input)
        fig, ax = plt.subplots(6, 3, figsize=(50, 50))
        for i in range(6):
            ax[i, 0].imshow((input[i].squeeze()))
            ax[i, 1].imshow((target[i].squeeze()), cmap=cmap)
            ax[i, 2].imshow((pred[i].squeeze()), cmap=cmap)

    else:
        fig, ax = plt.subplots(6, 2, figsize=(50, 50))
        for i in range(6):
            ax[i, 0].imshow((input[i].squeeze()))
            ax[i, 1].imshow((target[i].squeeze()), cmap=cmap)


visualize_samples = next(
    iter(DataGenerator(data=df, batch_size=6, dim=(HEIGHT, WIDTH)))
)
visualize_depth_map(visualize_samples)

Обратная трассировка ниже:

 StopIteration                             Traceback (most recent call last)
<ipython-input-7-1f95b664ead3> in <module>()
     19 
     20 
---> 21 visualize_samples = next(iter(DataGenerator(data=df, batch_size=6, dim=(HEIGHT, WIDTH))))
     22 visualize_depth_map(visualize_samples)

StopIteration:

Комментарии:

1. не могли бы вы вставить сюда обратную трассировку ошибки? что может помочь нам посмотреть

2. @TamilSelvan добавил трассировку

3. Какова форма вашего df фрейма данных?

4. Я имею в виду тот же набор данных и код, что и в keras.io/examples/vision/depth_estimation

5. Можете ли вы распечатать df.shape непосредственно перед применением next ?

Метки: