#python #tensorflow #conv-neural-network #tensorflow-lite
Вопрос:
Я использую внутри инициализации():
from tflite_runtime.interpreter import Interpreter
self.interpreter = Interpreter(self.selectedLayer)
Но я получаю ValueError: Mmap of '118' failed.
У меня есть массив доступных слоев, которые затем инициализируются при каждом запуске нового теста.
def __init__(self, layer):
availableLayers= {
"apps":'/home/pi/Desktop/python/NNs/tflite/appModel'
# ...more layers like this
}
self.layerChosen = layer
self.saveDirectory = "/home/pi/Desktop/"
self.selectedLayer = availableLayers[layer] "/model.tflite"
self.label = self.loadLabels(availableLayers[layer] "/labels.txt")
self.interpreter = Interpreter(self.selectedLayer)
# This is where the Mmap error is occuring
self.interpreter.allocate_tensors()
self.input_details = self.interpreter.get_input_details()
self.output_details = self.interpreter.get_output_details()
self.floating_model = self.input_details[0]['dtype'] == np.float32
self.height = self.input_details[0]['shape'][1]
self.width = self.input_details[0]['shape'][2]
Эта ошибка возникает после того, как уже выполнено несколько тестов. Это init()
повторяется при каждом новом тестировании.
Имея это в виду, я думаю, что это может быть проблема с памятью. Есть ли способ увеличить объем памяти до Interpretor
?