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})