Чтобы исправить ошибку, объект 'numpy .ndarray' не имеет атрибута 'convert' - PullRequest
Новая
       56

Чтобы исправить ошибку, объект 'numpy .ndarray' не имеет атрибута 'convert'

0 голосов
/ 06 мая 2020

Я применил деконволюцию изображения, используя неконтролируемый алгоритм Винера, и увеличил резкость и контрастность на определенном c наборе данных. Но при компиляции кода я столкнулся с ошибкой. Он показывает AttributeError: объект 'numpy .ndarray' не имеет атрибута 'convert'. Как это исправить? Мой код приведен ниже - 

import cv2
import glob
from matplotlib import pyplot as plt
from skimage import io, color, restoration, img_as_float
import scipy.stats as st
import numpy as np
from PIL import Image
from PIL import ImageEnhance

all_img = glob.glob('input/*.png')
other_dir = 'output/'

for img_id, img_path in enumerate(all_img):
    img = img_as_float(io.imread(img_path,0))

    def gkern(kernlen=21, nsig=2):   
        lim = kernlen//2 + (kernlen % 2)/2
        x = np.linspace(-lim, lim, kernlen+1)
        kern1d = np.diff(st.norm.cdf(x))
        kern2d = np.outer(kern1d, kern1d)
        return kern2d/kern2d.sum()

    psf = gkern(5,3)   

    deconvolved, _ = restoration.unsupervised_wiener(img, pdf)

    # Applied Sharpness and contrast
    enhancer_object = ImageEnhance.Contrast(deconvolved)
    out = enhancer_object.enhance(1.4)
    enhancer = ImageEnhance.Sharpness(out)
    enhanced_im = enhancer.enhance(8.0)  
    enhanced_cv_im = np.array(enhanced_im)

    cl2 = cv2.resize(enhanced_cv_im, (512,512), interpolation = cv2.INTER_CUBIC)
    plt.imsave(f"output/unsupervised_wiener_{img_id}.png", cl2, cmap='gray')

Он показывает ошибку - 

runfile('C:/Users/Junaed/.spyder-py3/unsupervised_wiener.py', wdir='C:/Users/Junaed/.spyder-py3')
Traceback (most recent call last):

  File "C:\Users\Junaed\.spyder-py3\unsupervised_wiener.py", line 37, in <module>
    enhancer_object = ImageEnhance.Contrast(deconvolved)

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

1 Ответ

1 голос
/ 06 мая 2020

ImageEnhance.Contrast () ожидает изображения PIL, на котором он может запускать image.convert, но вы передали ему массив numpy. Чтобы преобразовать в PIL, вы можете сделать это 

from PIL import Image
import numpy 

im = Image.fromarray(numpy.uint8(deconvolved))
...