Ошибка при загрузке изображения в обученную модель - PullRequest
0 голосов
/ 02 ноября 2019

Это ошибка, которую я получаю при загрузке тестового изображения для прогнозирования с помощью предварительно обученной модели. Пожалуйста, помогите!

tensorflow.python.framework.errors_impl.FailedPreconditionError: Error while reading resource variable conv2d_2/bias from Container: localhost. This could mean that the variable was uninitialized. Not found: Container localhost does not exist. (Could not find resource: localhost/conv2d_2/bias)
     [[{{node conv2d_2/BiasAdd/ReadVariableOp}}]]

Вот так я загружаю сохраненную модель и вызываю модель для прогнозирования по загруженному изображению

    graph = tf.get_default_graph()
    with graph.as_default():
        # load model at very first
        model = load_model(STATIC_FOLDER + '/' + '*********.h5')


    # call model to predict an image
    def api(full_path):
        data = image.load_img(full_path, target_size=(150, 150, 3))
        data = np.expand_dims(data, axis=0)
        data = data * 1.0 / 255

        with graph.as_default():
            predicted = model.predict(data)
            return predicted

Вот так я обрабатываю загруженный файл ипредсказывая это в Python.

    @app.route('/upload', methods=['POST','GET'])
    def upload_file():

        if request.method == 'GET':
            return render_template('index.html')
       else:
            file = request.files['image']
            full_name = os.path.join(UPLOAD_FOLDER, file.filename)
            file.save(full_name)

            indices = {0: 'Counterfeit', 1: 'Authentic'}
            result = api(full_name)

            predicted_class = np.asscalar(np.argmax(result, axis=1))
            accuracy = round(result[0][predicted_class] * 100, 2)
            label = indices[predicted_class]

        return render_template('predict.html', image_file_name = file.filename, label = label, accuracy = accuracy)

    @app.route('/uploads/<filename>')
    def send_file(filename):
        return send_from_directory(UPLOAD_FOLDER, filename)

Конфигурация системы:

Python               3.7.4
tensorflow           2.0.0    
tensorflow-estimator 2.0.1 
Keras                2.3.1    
Keras-Applications   1.0.8    
Keras-Preprocessing  1.1.0 
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...