Skip to content
Snippets Groups Projects
Commit 3862a0bd authored by tgupta6's avatar tgupta6
Browse files

disconnect word vectors and obj/atr classifiers

parent 5712f119
No related branches found
No related tags found
No related merge requests found
......@@ -34,18 +34,28 @@ class ObjectAttributeInference():
self.attribute_embed = self.add_attribute_graph(self.avg_pool_feat)
with tf.variable_scope('object_score_graph'):
self.object_scores = tf.matmul(
# self.object_scores = tf.matmul(
# self.object_embed,
# tf.transpose(self.object_label_vectors))
self.object_scores = layers.full(
self.object_embed,
tf.transpose(self.object_label_vectors))
constants.num_object_labels,
'obj_class_fc_layer',
func=None)
self.object_prob = tf.nn.softmax(
self.object_scores,
name = 'object_prob')
with tf.variable_scope('attribute_score_graph'):
self.attribute_scores = tf.matmul(
# self.attribute_scores = tf.matmul(
# self.attribute_embed,
# tf.transpose(self.attribute_label_vectors))
self.attribute_scores = layers.full(
self.attribute_embed,
tf.transpose(self.attribute_label_vectors))
constants.num_attribute_labels,
'atr_class_fc_layer',
func=None)
self.attribute_prob = tf.sigmoid(
self.attribute_scores,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment