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

模型更改

Browse Source
This commit is contained in:
xinyang
2019-08-15 16:39:03 +08:00
parent 7a05018cd1
commit 1163cf4c02
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tools/TrainCNN/backward.py
View File

@@ -52,8 +52,8 @@ def save_para(folder, paras, names, info):
STEPS = 100000
BATCH = 50
LEARNING_RATE_BASE = 0.0005
BATCH = 40
LEARNING_RATE_BASE = 0.0003
LEARNING_RATE_DECAY = 0.99
MOVING_AVERAGE_DECAY = 0.99
@@ -98,7 +98,7 @@ def train(dataset, show_bar=False):
_, loss_value, step = sess.run(
[train_op, loss, global_step],
feed_dict={x: images_samples, y_: labels_samples, keep_rate: 0.4}
feed_dict={x: images_samples, y_: labels_samples, keep_rate: 0.3}
)
if step % 500 == 0:

6
tools/TrainCNN/forward.py
View File

@@ -29,19 +29,19 @@ def max_pool_2x2(x):
CONV1_KERNAL_SIZE = 5
# 第一层卷积输出通道数
CONV1_OUTPUT_CHANNELS = 4
CONV1_OUTPUT_CHANNELS = 6
# 第二层卷积核大小
CONV2_KERNAL_SIZE = 3
# 第二层卷积输出通道数
CONV2_OUTPUT_CHANNELS = 8
CONV2_OUTPUT_CHANNELS = 10
# 第三层卷积核大小
CONV3_KERNAL_SIZE = 3
# 第三层卷积输出通道数
CONV3_OUTPUT_CHANNELS = 12
CONV3_OUTPUT_CHANNELS = 14
# 第一层全连接宽度
FC1_OUTPUT_NODES = 60