From d1b9e8b530656f6236ebe488de739c8a35d4ce4c Mon Sep 17 00:00:00 2001 From: xinyang Date: Fri, 3 May 2019 10:58:24 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DCNN=E8=AE=AD=E7=BB=83?= =?UTF-8?q?=E4=BB=A3=E7=A0=81=E7=9A=84=E5=B0=8FBUG?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools/TrainCNN/backward.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/TrainCNN/backward.py b/tools/TrainCNN/backward.py index 66f421c..9d21e0a 100755 --- a/tools/TrainCNN/backward.py +++ b/tools/TrainCNN/backward.py @@ -103,7 +103,7 @@ def train(dataset, show_bar=False): if i % 100 == 0: if i % 1000 == 0: - test_samples, test_labels = dataset.sample_test_sets(5000) + test_samples, test_labels = dataset.sample_test_sets(5000) acc = sess.run(accuracy, feed_dict={x: test_samples, y_: test_labels}) bar.set_postfix({"loss": loss_value, "acc": acc}) @@ -142,8 +142,8 @@ def train(dataset, show_bar=False): # res = sess.run(y, feed_dict={x: im}) # res = res.reshape([forward.OUTPUT_NODES]) # print(np.argmax(res)) - - test_samples, test_labels = dataset.sample_test_sets(100) + + test_samples, test_labels = dataset.sample_test_sets(100) vars_val = sess.run(vars) save_para("/home/xinyang/Desktop/AutoAim/tools/para", vars_val) nodes_val = sess.run(nodes, feed_dict={x:test_samples})