我正在尝试实施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)