отображать несколько изображений в одном каталоге с помощью imshow Python - PullRequest
Новая
       19

отображать несколько изображений в одном каталоге с помощью imshow Python

0 голосов
/ 22 апреля 2019

Я пытаюсь отобразить несколько изображений в одном каталоге. 

import os
import cv2
import numpy as np
import random
import math

import matplotlib
from matplotlib.pyplot import imshow
from matplotlib import pyplot as plt
import matplotlib.image as mpimg
%matplotlib inline

dir_path = 'img'
images = os.listdir(dir_path)
img_paths = [os.path.join(dir_path, i) for i in images]
img_paths.sort()
img_all = np.array([cv2.cvtColor(cv2.imread(p), cv2.COLOR_BGR2RGB) for p in img_paths])


def display_helper(images, cmap=None):
    fig, ax = plt.subplots(nrows=20, ncols=2, figsize=(15,6))
    for a in ax:
        a.imshow(img, interpolation='none')

display_helper(img_all)

Но я получаю эту ошибку 

AttributeError: 'numpy.ndarray' object has no attribute 'imshow'

Как отобразить несколько изображений с помощью imshow?

Спасибо!

1 Ответ

0 голосов
/ 23 апреля 2019

Проблема здесь 

fig, ax = plt.subplots(nrows=20, ncols=2, figsize=(15,6))

, которая дает ax как np.array формы (20,2).Поэтому, когда вы делаете for a in ax, a - это np.array из 2 осей.Чтобы исправить это, измените следующую строку на: 

for a in ax.flatten():
...