即使是过度拟合的情况,平均iOU也很低

时间:2019-04-04 21:32:15

标签: tensorflow deep-learning semantic-segmentation

我正在尝试实施FCN网络。为了评估我的结果,我添加了平均IOU,但是,如果在小型数据集上过度拟合训练网络,则我的平均IOU会非常低,尽管损失非常低。我请任何向导知道我在做什么错。这是我正在使用估算器运行的代码。

iou_weights = tf.cast(tf.less_equal(annotation, num_cls - 1),
                  tf.int32)  # Ignoring all labels greater than or equal to   n_classes.The last class is ignored class.

weighted_miou, weighted_update_op = tf.contrib.metrics.streaming_mean_iou(labels=annotation,
                                                                          predictions=pred_annotation,
                                                                          num_classes=num_cls,
                                                                            weights=iou_weights)
with tf.control_dependencies([tf.identity(weighted_update_op)]):
    weighted_miou = tf.identity(weighted_miou)

0 个答案:

没有答案