leeyoshinari / YOLO_v2

The implementation of YOLO v2 with TensorFlow
GNU General Public License v3.0
80 stars 38 forks source link

why your loss is very low #5

Closed LPaKing closed 6 years ago

leeyoshinari commented 6 years ago

Because I use reduce_mean, change loss = tf.reshape(loss, [self.batch_size, self.cell_size * self.cell_size * self.box_per_cell * (self.num_class + 5)]) and loss = 0.5 * tf.reduce_mean(tf.reduce_mean(loss, axis = 1)) to loss = tf.reduce_mean(tf.reduce_sum(loss, axis = [1, 2, 3, 4]), name = 'loss'), the loss is a little bigger.

shaoweistar commented 6 years ago

what is the differents between this two loss

leeyoshinari commented 6 years ago

I think there is no difference between them.