Как увеличить все изображения в папке с помощью TensorFlow - PullRequest
Новая
       72

Как увеличить все изображения в папке с помощью TensorFlow

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

Чтобы загрузить папку в colab и увеличить изображения

1 Ответ

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

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

import keras
import cv2
import os
import glob
from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
datagen = ImageDataGenerator(rotation_range =15, 
                         width_shift_range = 0.2, 
                         height_shift_range = 0.2,  
                         rescale=1./255, 
                         shear_range=0.2, 
                         zoom_range=0.2, 
                         horizontal_flip = True, 
                         fill_mode = 'nearest', 
                         data_format='channels_last', 
                         brightness_range=[0.5, 1.5]) 


img_dir = "folder-name" # Enter Directory of all images 
data_path = os.path.join(img_dir,'*g')
files = glob.glob(data_path)
data = []
for f1 in files:
    img = cv2.imread(f1)
    data.append(img)

x = img_to_array(img)
x = x.reshape((1,) + x.shape)

i = 0
path, dirs, files = next(os.walk("folder-name"))
file_count = len(files) #to find number of files in folder

for batch in datagen.flow (x, batch_size=1, save_to_dir =r'new-folder-name',save_prefix="a",save_format='jpg'):
    i+=1
    if i==file_count:
      break

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

...