更新CNN参数

tools/TrainCNN/backward.py

@@ -54,7 +54,7 @@ def save_para(folder, paras):
         save_bias(fp, paras[7])
 
 
-STEPS = 20000
+STEPS = 100000
 BATCH = 30
 LEARNING_RATE_BASE  = 0.01
 LEARNING_RATE_DECAY = 0.99
@@ -152,7 +152,7 @@ def train(dataset, show_bar=False):
 
 if __name__ == "__main__":
     print("Loading data sets...")
-    dataset = generate.DataSet("/home/xinyang/Desktop/DataSets/box")
+    dataset = generate.DataSet("/home/xinyang/Desktop/DataSets")
     print("Finish!")
     train(dataset, show_bar=True)
     input("Press any key to end...")

tools/TrainCNN/forward.py

@@ -29,20 +29,19 @@ def max_pool_2x2(x):
 CONV1_KERNAL_SIZE = 5
 
 # 第一层卷积输出通道数
-CONV1_OUTPUT_CHANNELS = 6
+CONV1_OUTPUT_CHANNELS = 8
 
 # 第二层卷积核大小
 CONV2_KERNAL_SIZE = 3
 
 # 第二层卷积输出通道数
-CONV2_OUTPUT_CHANNELS = 10
+CONV2_OUTPUT_CHANNELS = 16
 
 # 第一层全连接宽度
 FC1_OUTPUT_NODES = 16
 
 # 第二层全连接宽度（输出标签类型数）
-FC2_OUTPUT_NODES = 8
-
+FC2_OUTPUT_NODES = 11
 # 输出标签类型数
 OUTPUT_NODES = FC2_OUTPUT_NODES