Я хочу изменить размер каждого изображения до (32,128,1) и нормализовать его, чтобы передать его на мой CNN - PullRequest
Новая
       14

Я хочу изменить размер каждого изображения до (32,128,1) и нормализовать его, чтобы передать его на мой CNN

0 голосов
/ 08 октября 2019

Вот мой существующий код, который выдает ошибку, не возвращая четырехмерный ввод. Вот код. Есть ли способ, которым я могу его изменить? 

    img = cv2.cvtColor(cv2.imread(os.path.join(root, f_name)), cv2.COLOR_BGR2GRAY)   

    # convert each image of shape (32, 128, 1)
    w, h = img.shape
    if h > 128 or w > 32:
        continue
    if w < 32:
        add_zeros = np.ones((32-w, h))*255
        img = np.concatenate((img, add_zeros))

    if h < 128:
        add_zeros = np.ones((32, 128-h))*255
        img = np.concatenate((img, add_zeros), axis=1)
    img = np.expand_dims(img , axis = 2)

    # Normalize each image
    img = img/255.
...