Skip to content

S
seminar-breakout
Commit 20973118 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

bug fix in hed

parent 9c6eb092
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/HED/hed.py
View file @ 20973118
...@@ -22,6 +22,7 @@ class Model(ModelDesc): ...@@ -22,6 +22,7 @@ class Model(ModelDesc):
def _build_graph(self, input_vars): def _build_graph(self, input_vars):
image, edgemap = input_vars image, edgemap = input_vars
image = image - tf.constant([104, 116, 122], dtype='float32') image = image - tf.constant([104, 116, 122], dtype='float32')
edgemap = tf.expand_dims(edgemap, 3)
def branch(name, l, up): def branch(name, l, up):
with tf.variable_scope(name) as scope: with tf.variable_scope(name) as scope:
...@@ -70,7 +71,7 @@ class Model(ModelDesc): ...@@ -70,7 +71,7 @@ class Model(ModelDesc):
for idx, b in enumerate([b1, b2, b3, b4, b5, final_map]): for idx, b in enumerate([b1, b2, b3, b4, b5, final_map]):
output = tf.nn.sigmoid(b, name='output{}'.format(idx+1)) output = tf.nn.sigmoid(b, name='output{}'.format(idx+1))
xentropy = class_balanced_sigmoid_cross_entropy( xentropy = class_balanced_sigmoid_cross_entropy(
tf.squeeze(b, [3]), edgemap, b, edgemap,
name='xentropy{}'.format(idx+1)) name='xentropy{}'.format(idx+1))
costs.append(xentropy) costs.append(xentropy)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment