def _area_loss(pred_bbox, img_size, presence):
area = pred_bbox[..., 2] * pred_bbox[..., 3]
ratio = area / tf.reduce_prod(tf.to_float(img_size))
weights = tf.clip_by_value(ratio, 1., 10.)
ratio = tf.clip_by_value(ratio, 0., 1.)
return _time_weighted_nll(1 - ratio, presence, weights)
评论列表
文章目录