VSAG/vision_sjtu_19
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复CNN训练代码的小BUG

Browse Source
This commit is contained in:
xinyang
2019-05-03 10:58:24 +08:00
parent 174809e6a7
commit d1b9e8b530
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tools/TrainCNN/backward.py
View File

@@ -103,7 +103,7 @@ def train(dataset, show_bar=False):
if i % 100 == 0: if i % 100 == 0:
if i % 1000 == 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}) acc = sess.run(accuracy, feed_dict={x: test_samples, y_: test_labels})
bar.set_postfix({"loss": loss_value, "acc": acc}) 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 = sess.run(y, feed_dict={x: im})
# res = res.reshape([forward.OUTPUT_NODES]) # res = res.reshape([forward.OUTPUT_NODES])
# print(np.argmax(res)) # 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) vars_val = sess.run(vars)
save_para("/home/xinyang/Desktop/AutoAim/tools/para", vars_val) save_para("/home/xinyang/Desktop/AutoAim/tools/para", vars_val)
nodes_val = sess.run(nodes, feed_dict={x:test_samples}) nodes_val = sess.run(nodes, feed_dict={x:test_samples})