center loss delete

rockyzhengwu · Oct 27, 2019 · 870467d · 870467d
1 parent 82ee67f
commit 870467d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/single_word_ocr/densenet.py b/single_word_ocr/densenet.py
@@ -22,9 +22,9 @@ def __init__(self, batch_size, num_classes, mode='train', center_loss_alpha=0.95
 
     if self.is_training:
       print("feautres =====> ", self.features)
-      self.center_loss, _ = self.center_loss(self.features, self.labels, self.center_loss_alpha, self.num_classes)
-      self.softmax_loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(logits=self.logits, labels=self.labels))
-      self.loss = self.softmax_loss + self.center_loss
+      #self.center_loss, _ = self.center_loss(self.features, self.labels, self.center_loss_alpha, self.num_classes)
+      #self.softmax_loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(logits=self.logits, labels=self.labels))
+      self.loss = self.softmax_loss 
 
     self.predict_prob = tf.nn.softmax(self.logits, name='prob' )
     print("predict_prob====>", self.predict_prob)

diff --git a/single_word_ocr/train.py b/single_word_ocr/train.py
@@ -38,8 +38,8 @@ def train():
   #optimizer=tf.train.MomentumOptimizer(learning_rate=learning_rate, momentum=0.9).minimize(loss=model.loss)
   saver = tf.train.Saver()
   tf.summary.scalar(name='loss', tensor=model.loss)
-  tf.summary.scalar(name='softmax_loss', tensor=model.softmax_loss)
-  tf.summary.scalar(name='center_loss', tensor=model.center_loss)
+  #tf.summary.scalar(name='softmax_loss', tensor=model.softmax_loss)
+  #tf.summary.scalar(name='center_loss', tensor=model.center_loss)
   tf.summary.scalar(name='accuracy', tensor=model.accuracy)
   merge_summary_op = tf.summary.merge_all()
   sess_config = tf.ConfigProto(allow_soft_placement=True,)