全套代码可至本人github下载地址
使用步骤:
下面是我自己写的测试代码,上面地址下载的也会包含
# 输入:测试集标注文件夹和对应的预测文件夹,图片命名为 标注+x;predict+x
# 输出:四个指标的平均值
from eval_segm import *
import cv2 as cv
def threshold_demo(a):
gray = cv.cvtColor(a, cv.COLOR_RGB2GRAY) # 把输入图像灰度化
# 直接阈值化是对输入的单通道矩阵逐像素进行阈值分割。
ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_TRIANGLE)
print("threshold value %s"%ret)
return binary
biaozhu_path = xxx
predict_path = xxx
pa = []
mpa = []
miou = []
fwiou = []
number_photo = 1
for i in range(number_photo):
i = i+1
a = cv.imread(biaozhu_path + 'biaozhu' + str(i) + '.png')
b = cv.imread(predict_path