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-06 11:54:48 +08:00
parent 7337f10123
commit 9392c201db
20 changed files with 41259 additions and 17856 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tools/TrainCNN/forward.py
View File

@@ -35,16 +35,16 @@ CONV1_OUTPUT_CHANNELS = 4
CONV2_KERNAL_SIZE = 3
# 第二层卷积输出通道数
CONV2_OUTPUT_CHANNELS = 6
CONV2_OUTPUT_CHANNELS = 8
# 第三层卷积核大小
CONV3_KERNAL_SIZE = 3
# 第三层卷积输出通道数
CONV3_OUTPUT_CHANNELS = 8
CONV3_OUTPUT_CHANNELS = 16
# 第一层全连接宽度
FC1_OUTPUT_NODES = 50
FC1_OUTPUT_NODES = 60
# 第二层全连接宽度（输出标签类型数）
FC2_OUTPUT_NODES = 15
@@ -62,8 +62,8 @@ def forward(x, regularizer=None, keep_rate=tf.constant(1.0)):
[CONV1_KERNAL_SIZE, CONV1_KERNAL_SIZE, int(x.shape[3]), CONV1_OUTPUT_CHANNELS]
)
conv1_b = get_bias([CONV1_OUTPUT_CHANNELS])
conv1 = tf.nn.relu(tf.nn.bias_add(conv2d(x, conv1_w), conv1_b))
pool1 = avg_pool_2x2(conv1)
conv1 = tf.nn.relu(tf.nn.bias_add(conv2d(x, conv1_w), conv1_b))
pool1 = avg_pool_2x2(conv1)
print("conv1: ", conv1.shape)
print("pool1: ", pool1.shape)
vars.extend([conv1_w, conv1_b])
@@ -100,7 +100,7 @@ def forward(x, regularizer=None, keep_rate=tf.constant(1.0)):
fc1_w = get_weight([node, FC1_OUTPUT_NODES], regularizer)
fc1_b = get_bias([FC1_OUTPUT_NODES])
fc1 = tf.nn.relu(tf.matmul(reshaped, fc1_w) + fc1_b)
fc1 = tf.nn.relu(tf.matmul(reshaped, fc1_w) + fc1_b)
vars.extend([fc1_w, fc1_b])
vars_name.extend(["fc1_w", "fc1_b"])
nodes.extend([fc1])
@@ -113,4 +113,3 @@ def forward(x, regularizer=None, keep_rate=tf.constant(1.0)):
nodes.extend([fc2])
return nodes, vars, vars_name