资源简介

# 执行命令: python miou.py --labels=D:/.../labels --pred=D:/../infer # 其中 --labels 为标注图片路径 --pred 为预测出的图片路径 # 最终结果会打印输出,也会在同级目录下,生成一个txt文档,保存各个类别 iou # 操作极为简单

资源截图

代码片段和文件信息

# 执行命令:
# python miou.py --labels=D:/.../labels --pred=D:/../infer
# 其中  --labels 为标注图片路径    --pred 为预测出的图片路径
# 会在同级目录下,生成一个txt文档,保存各个类别 iou


import argparse
import glob
import imageio
import numpy as np
import os
import tensorflow as tf
import time
import pandas as pd
import seaborn as sn
import matplotlib.pyplot as plt
from matplotlib import gridspec
from datetime import datetime
from tensorflow.python.client import device_lib
from socket import gethostname
import itertools



def _calc_metrics(predictions labels n_classes):
    with tf.variable_scope(“metrics“):
        mean_iou mean_iou_op = tf.metrics.mean_iou(
            labels=labels
            predictions=predictions
            num_classes=n_classes
            name=‘my_miou‘)
        conf_m

评论

共有 条评论