как справиться с разным размером изображений в сети CNN - PullRequest
Новая
       64

как справиться с разным размером изображений в сети CNN

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

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

solution.head()
id  pair
0   1   686
1   2   540
2   3   971
3   4   910
4   5   616

У меня есть изображения, подобные этим: 

    png = []
    for image_path in glob.glob("training/training/*.png"):
        png.append(misc.imread(image_path))    

    im = np.asarray(png)

    print ('Importing done...', im.shape)

Importing done... (1000,)

array([array([[[247, 255, 102],
        [247, 255, 102],
        [247, 255, 102],
        ...,
        [247, 255, 102],
        [247, 255, 102],
        [247, 255, 102]],

       [[247, 255, 102],
        [247, 255, 102],
        [247, 255, 102],
        ...,
        [247, 255, 102],
        [247, 255, 102],
        [247, 255, 102]],

       [[247, 255, 102],
        [247, 255, 102],
        [247, 255, 102],
        ...,
        [247, 255, 102],
        [247, 255, 102],
        [247, 255, 102]],

       ...,

       [[247, 255, 102],
        [247, 255, 102],
        [247, 255, 102],
        ...,
        [247, 255, 102],
        [247, 255, 102],
        [247, 255, 102]],

       [[247, 255, 102],
        [247, 255, 102],
        [247, 255, 102],
        ...,
        [247, 255, 102],
        [247, 255, 102],
        [247, 255, 102]],

       [[247, 255, 102],
        [247, 255, 102],
        [247, 255, 102],
        ...,
        [247, 255, 102],
        [247, 255, 102],
        [247, 255, 102]]], dtype=uint8),
       array([[[165, 255, 102],
        [165, 255, 102],
        [165, 255, 102],
        ...,
        [165, 255, 102],
        [165, 255, 102],
        [165, 255, 102]],

       [[165, 255, 102],
        [165, 255, 102],
        [165, 255, 102],
        ...,
        [165, 255, 102],
        [165, 255, 102],
        [165, 255, 102]],

       [[165, 255, 102],
        [165, 255, 102],
        [165, 255, 102],
        ...,
        [165, 255, 102],
        [165, 255, 102],
        [165, 255, 102]],

       ...,

       [[165, 255, 102],
        [165, 255, 102],
        [165, 255, 102],
        ...,

Я не понимал, как найти идеальные пары изображений. Я пытаюсь изменить их размер и преобразовать в серый, но получаю ошибку. Пожалуйста, помогите мне найти изображения, которые можно объединить в пары. вот моя ссылка на колаб: https://colab.research.google.com/drive/1bCQUHrXNjdWX8-WKSq6LK01XpR4D52J_

1 Ответ

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

Вы можете попробовать этот код: 

   import numpy as np
    import cv2
    import os
    df=pd.read_csv(os.path.join(file_path,'solution.csv'))
    training_data = []
    IMG_SIZE = 100
    for img in  os.listdir('training/training') :  
      path = 'training/training/'+img
      id = img.split('.')[0]
      img = cv2.imread(path,1)
      img = cv2.resize(img, (IMG_SIZE,IMG_SIZE),3)     
      img = np.array(img)
      target = df['pair][int(id)]
      training_data.append([img ,target  ])
...