Привет У меня проблемы с запуском модели 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, но ничего не было исправлено.
Пожалуйста, помогите.