更新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
 

tools/para/conv1_b

@@ -1,7 +1,9 @@
-6
+8
-0.25190517
+1.4952865
--0.064696774
+0.71153855
--1.1652193
+-3.1710217
--1.3264673
+-0.13555647
-2.689583
+-0.1106117
--1.1748564
+-0.67361253
+0.8526688
+-0.5431338

tools/para/conv2_b

@@ -1,11 +1,17 @@
-10
+16
-2.339627
+-8.03951
-1.8759418
+3.4012065
--0.70012397
+-0.13434531
--0.13009475
+-2.9315674
-0.17396869
+-0.19703378
-2.766548
+-1.5522652
--1.4850869
+-2.032869
--2.6339962
+-3.2706444
--0.12728444
+-0.21717867
-0.14405365
+-0.19119318
+-0.15644628
+8.05552
+-1.1173456
+1.5701971
+-0.081088655
+-0.1808035

tools/para/fc1_b

@@ -1,17 +1,17 @@
 16
--1.0079371
+-3.9457524
--0.21348575
+-0.3353631
--0.047885668
+-0.53903425
-1.3191578
+-0.22991405
--0.18780966
+8.231101
-7.9113836
+-0.29107538
--0.07285687
+-0.20410632
--4.019421
+-0.9607706
--0.9552737
+-0.56446344
--0.41599488
+-0.17112184
--0.13497822
+29.32642
--1.286887
+7.855335
-1.954279
+-2.8522294
--0.377049
+-0.097452246
--0.2319671
+1.2676659
--5.739259
+-0.038555548

tools/para/fc2_b

@@ -1,7 +1,12 @@
-6
+11
-7.672486
+4.68946
--2.1727772
+2.4254043
--1.6976244
+-1.1937128
--0.39480045
+12.141608
--4.701963
+-9.4490185
--4.060234
+2.2703283
+-5.419338
+-7.4511914
+0.31088707
+-2.521978
+-4.9146705

tools/para/fc2_w

@@ -1,98 +1,178 @@
 16
-6
+11
-5.294904e-34
+-0.01987104
-4.1071474e-34
+0.42782968
-3.5979983e-34
+-0.12000261
--4.8111744e-34
+-0.22871323
-3.0298388e-34
+5.909024e-06
-4.727617e-34
+0.12846695
-4.2841314e-34
+0.34395957
--7.448757e-34
+-0.025282092
--1.3280983e-34
+-0.2352891
--3.4998216e-34
+0.060300585
-1.1480016e-34
+-0.11231345
-2.4329347e-34
+-7.6262363e-34
--3.8421887e-34
+-1.1019437e-33
-6.131607e-34
+2.395344e-34
-3.2127673e-35
+-7.14328e-34
-4.036833e-34
+1.12534335e-33
--4.4449285e-34
+3.6429394e-34
--1.3033124e-34
+-1.09642895e-33
--0.0063951695
+-5.5404825e-35
-0.006757953
+-1.2785976e-34
-0.026418086
+4.6425433e-34
--0.014205176
+2.0518196e-34
-0.04418388
+-3.817569e-34
-0.062677905
+-1.3871395e-08
--5.2201886e-34
+-8.454291e-34
-1.1684014e-33
+6.484888e-34
--4.5119606e-34
+-9.504644e-12
-5.0112353e-34
+3.8818015e-34
-7.1335205e-34
+4.3159157e-34
-3.8090283e-34
+-7.718197e-34
-0.61706346
+5.7451056e-34
-0.12671919
+-4.4890632e-34
--0.05489751
+-3.9756725e-34
--0.03665739
+7.855176e-34
--0.2387121
+1.04599e-33
--0.19392832
+-1.2205084e-33
-4.400894e-34
+-2.641615e-34
-3.3940071e-34
+6.133725e-34
--3.2267908e-34
+3.8102495e-34
-4.2374877e-34
+-9.052317e-34
-3.7673267e-34
+-1.1754962e-33
--4.2796806e-34
+8.280429e-34
-0.09789613
+7.856483e-34
--0.37246978
+-2.1041588e-34
--0.09926232
+0.0044932007
-0.054989994
+-0.3508491
-0.5217698
+-0.4295554
--0.020826261
+-0.13176644
-4.7286482e-15
+9.088804e-06
--1.3830426e-15
+-0.35067832
-5.8647296e-35
+0.37535986
-1.1566057e-33
+0.28786227
-1.2851214e-33
+0.44722718
--4.4416296e-33
+0.24968775
-2.4679793e-34
+0.31687224
--8.546679e-34
+9.918349e-34
--6.019269e-34
+-1.3927404e-33
--4.7928078e-34
+-1.2877562e-33
--2.6198876e-35
+1.0205193e-33
--1.5769217e-36
+9.401986e-34
--1.0960844e-33
+-5.9421037e-34
--1.1464184e-33
+-1.1976041e-33
--4.9895474e-34
+-1.4003625e-33
--1.2327797e-33
+-2.3580773e-36
-6.3556863e-34
+7.481718e-34
-3.949529e-36
+4.8942575e-34
--0.44523916
+4.6177516e-35
--0.048138373
+-6.6717886e-34
-0.49848852
+3.223957e-34
-0.67901206
+-1.584663e-34
--0.035957992
+6.743674e-34
--0.36799058
+1.6929226e-34
-0.07924421
+-1.6124294e-34
--0.23662093
+-4.8999394e-34
-0.409469
+3.5384667e-34
--0.60837746
+4.6176213e-35
--0.14665303
+-4.3959643e-36
-0.685228
+4.8607646e-34
--2.1080804e-34
+-2.2389736e-07
-8.089918e-34
+-1.0547912e-34
--3.8773814e-34
+4.7272058e-35
--1.2162282e-34
+2.1540245e-09
-3.8192135e-34
+1.0148082e-33
-5.806797e-34
+-1.2641985e-34
-1.9141679e-34
+-4.718617e-34
-5.46885e-34
+-2.9784954e-34
--9.801028e-34
+-9.195764e-34
-8.670833e-34
+-5.4015044e-35
-2.838885e-34
+-9.436068e-34
-1.2663823e-34
+2.340125e-08
--0.21810062
+-6.886075e-34
-0.7242532
+6.8557177e-34
--0.07114903
+-2.3418155e-11
--0.066113256
+-8.925727e-34
--0.054350372
+-2.6056772e-34
--0.18786566
+-2.770205e-35
+-1.3465729e-33
+-2.8787352e-35
+2.5913718e-34
+-2.7904087e-34
+-4.580196e-34
+4.9749566e-34
+7.079356e-34
+1.5309345e-34
+5.076401e-34
+-7.8047275e-35
+5.7707056e-34
+1.0374484e-33
+-7.2734595e-34
+-2.4512414e-34
+0.5898928
+-0.13292696
+0.13457516
+-0.3083663
+-4.5226676e-05
+-0.016318614
+-0.056287136
+-0.0416729
+-0.13633117
+-0.034681365
+-0.058179155
+-0.053308636
+-0.15685758
+-0.057880845
+-0.049490057
+-4.48231e-06
+0.54238904
+-0.12667982
+-0.11757153
+-0.2049767
+-0.12322974
+0.46258917
+0.3563643
+-0.0011422047
+0.012159694
+0.10425257
+7.641402e-06
+-0.0068169823
+-0.061314583
+-0.20332839
+-0.19299586
+0.43478948
+-0.006821075
+5.7795678e-34
+-9.393816e-35
+9.3423295e-34
+-4.974839e-34
+-1.2766868e-33
+6.1611773e-34
+-1.0581506e-34
+-8.838601e-34
+-1.4182854e-33
+-7.169634e-34
+6.2219504e-34
+-0.05864723
+0.012660546
+0.47422087
+-0.12896037
+8.181692e-05
+0.15611899
+-0.043533176
+0.39083794
+-0.4173265
+-0.02371774
+-0.0947099
+7.7225295e-35
+-1.7053759e-23
+-5.490969e-23
+-2.3247693e-23
+-1.0633658e-22
+9.971754e-23
+4.4417918e-24
+5.392504e-23
+2.8454636e-23
+8.56064e-23
+9.8926345e-23