Если операторы не работают с переменными Tensorflow

Я попытался запрограммировать машину для самостоятельного вождения на Python и Tensorflow с помощью этого кода:

import tensorflow as tf
from PIL import ImageGrab
import numpy as np
import pyautogui as pag
import time

x_p = tf.placeholder(tf.uint8)
y_p = tf.placeholder(tf.float32)

weights = [tf.Variable(tf.random_normal([5,5,3,32],0.1)),
           tf.Variable(tf.random_normal([5,5,32,64],0.1)),
           tf.Variable(tf.random_normal([5,5,64,128],0.1)),
           tf.Variable(tf.random_normal([25*25*128,1064],0.1)),
           tf.Variable(tf.random_normal([1064,1],0.1))]

def CNN(x, weights):
    output = tf.nn.conv2d(x, weights[0], [1,1,1,1], 'SAME')
    output = tf.nn.relu(output)
    output = tf.nn.conv2d(output, weights[1], [1,2,2,1], 'SAME')
    output = tf.nn.relu(output)
    output = tf.nn.conv2d(output, weights[2], [1,2,2,1], 'SAME')
    output = tf.nn.relu(output)
    output = tf.reshape(output, [-1,25*25*128])
    output = tf.matmul(output, weights[3])
    output = tf.nn.relu(output)
    output = tf.matmul(output, weights[4])
    output = tf.reduce_sum(output)
    return output

prediction = CNN(tf.cast(x_p, tf.float32), weights)
saver = tf.train.Saver()
init = tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init)
    saver.restore(sess, 'saved/model.ckpt')
    for t in range(5):
        print(t+1)
        time.sleep(1)

    while True:
        x = ImageGrab.grab()
        x = x.resize((100,100))
        x = np.asarray(x)
        output = sess.run(prediction, feed_dict={x_p:[x]})
        print(output)
        if output < 0.5 and output > 1.5:
            pag.keyDown('W')
            pag.keyUp('S')
            pag.keyUp('D')
            pag.keyUp('A')
        elif output < 1.5 and output > 2.5:
            pag.keyUp('W')
            pag.keyDown('S')
            pag.keyUp('D')
            pag.keyUp('A')
        elif output < 2.5 and output > 3.5:
            pag.keyDown('W')
            pag.keyUp('S')
            pag.keyDown('D')
            pag.keyUp('A')
        elif output < 3.5 and output > 4.5:
            pag.keyDown('W')
            pag.keyUp('S')
            pag.keyUp('D')
            pag.keyDown('A')
        elif output < 4.5 and output > 5.5:
            pag.keyDown('W')
            pag.keyUp('S')
            pag.keyDown('D')
            pag.keyUp('A')
        elif output < 5.5 and output > 6.5:
            pag.keyDown('W')
            pag.keyUp('S')
            pag.keyUp('D')
            pag.keyDown('A')
        elif output < 6.5 and output > 7.5:
            pag.keyUp('W')
            pag.keyDown('S')
            pag.keyDown('D')
            pag.keyUp('A')
        elif output < 7.5 and output > 8.5:
            pag.keyUp('W')
            pag.keyDown('S')
            pag.keyUp('D')
            pag.keyDown('A')
        else:
            pag.keyUp('W')
            pag.keyUp('S')
            pag.keyUp('D')
            pag.keyUp('A')

Но проблема в том, что срабатывает только оператор else, даже если output имеет (например) значение 1.3. Мне удалось выяснить, что проблема вызвана выводом переменной в операторах if, но мне не удалось ее решить.

1 ответ

У вас проблемы с заявлением if.

 if output < 0.5 and output > 1.5:

ваш результат не может быть ниже 0,5 и в то же время выше 1,5. Так что утверждение if все время неверно.

Проблема одинакова для всех остальных элифов

Я думаю, что вы хотите написать:

 if output > 0.5 and output < 1.5:
Другие вопросы по тегам