-
大小: 4KB文件類型: .py金幣: 2下載: 1 次發(fā)布日期: 2021-01-10
- 語(yǔ)言: Python
- 標(biāo)簽: 計(jì)算miou??tensorflow??miou代碼??
資源簡(jiǎn)介
# 執(zhí)行命令:
python miou.py --labels=D:/.../labels --pred=D:/../infer
# 其中 --labels 為標(biāo)注圖片路徑 --pred 為預(yù)測(cè)出的圖片路徑
# 最終結(jié)果會(huì)打印輸出,也會(huì)在同級(jí)目錄下,生成一個(gè)txt文檔,保存各個(gè)類別 iou
# 操作極為簡(jiǎn)單
代碼片段和文件信息
#?執(zhí)行命令:
#?python?miou.py?--labels=D:/.../labels?--pred=D:/../infer
#?其中??--labels?為標(biāo)注圖片路徑????--pred?為預(yù)測(cè)出的圖片路徑
#?會(huì)在同級(jí)目錄下,生成一個(gè)txt文檔,保存各個(gè)類別?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
評(píng)論
共有 條評(píng)論