Raspberry Pi Tensorflow / Keras Ошибка: выделение 67108864 превышает 10% системной памяти - PullRequest
0 голосов
/ 02 ноября 2018

Привет У меня проблемы с запуском модели ML на основе вышеупомянутого на Pi 3 Код:

import pandas as pd

import tensorflow as tf
# deep learning library. Tensors are just multi-dimensional arrays
import numpy as np


import time
import skimage
from PIL import Image
import numpy as np
import sys
import os
import csv

# Import the `transform` module from `skimage`
from skimage import transform 

from skimage import data

import glob
from os import walk


fileLists = []
image28 = []
value = []

format='.jpg'

path = '/testpic'
folder = os.fsencode(path)


for root, directories, filenames in os.walk(path):


  for filename in filenames: 
    print (os.path.join(root,filename)) 
    if filename.endswith(('.jpg', '.png', '.gif')):

        fileLists.append(skimage.data.imread(os.path.join(root,filename)))


   time1 = time.time()        
   images28  = [transform.resize(fileList, (227, 227)) for fileList in 
   fileLists]

   train_x = np.array(images28)

   new_model = tf.keras.models.load_model('ml.h5')

   predictions = new_model.predict(train_x)


   for i in range(10):

time1 = time.time()

print(np.argmax(predictions[i]))

time2 = time.time()

print((time2-time1))

Когда я запускаю его, я получаю следующую ошибку:

Использование бэкэнда TensorFlow. 2018-11-02 09: 43: 34.904211: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph 2018-11-02 09: 43: 34.937663: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph 2018-11-02 09: 43: 42.740079: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph 2018-11-02 09: 43: 43.125458: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph 2018-11-02 09: 43: 43.720775: W tenorflow / core / framework / allocator.cc: 122] Выделение 67108864 превышает 10% системной памяти. 2018-11-02 09: 43: 44.152943: W tenorflow / core / framework / allocator.cc: 122] Выделение 67108864 превышает 10% системной памяти. 2018-11-02 09: 43: 44.298756: W tenorflow / core / framework / allocator.cc: 122] Выделение 67108864 превышает 10% системной памяти. 2018-11-02 09: 43: 44.299426: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph 2018-11-02 09: 44: 08.801283: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph 2018-11-02 09: 44: 09.582225: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph 2018-11-02 09: 44: 09.788274: W tenorflow / core / framework / allocator.cc: 122] Выделение 67108864 превышает 10% системной памяти. 2018-11-02 09: 44: 10.183454: W tenorflow / core / framework / allocator.cc: 122] Выделение 67108864 превышает 10% системной памяти. 2018-11-02 09: 44: 28.074020: I tenorflow / core / grappler / optimizers / meta_optimizer.cc: 344] Запуск оптимизации для элемента захвата: tf_graph

После чего программа убивается.

В процессе обучения я изменил размер партии на 1, но ничего не было исправлено.

Пожалуйста, помогите.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...