使用python計算兩張圖片的dice指數
阿新 • • 發佈:2020-12-30
使用python計算兩張圖片的dice指數
import cv2
import os
import numpy as np
def dice(x,y):
s2 = cv2.imread(x, 0)# 模板
row, col = s2.shape[0], s2.shape[1]
d = []
s1 = cv2.imread(y, 0)
s = []
for r in range(row - 10):
for c in range(col - 10):
if s1[r][c] == s2[r][c]: # 計算影象畫素交集
s.append(s1[r][c])
# print(s1[r][c])
m1 = np.linalg.norm(s)
m2 = np.linalg.norm(s1.flatten()) + np.linalg.norm(s2.flatten())
d.append(2*m1/m2)
return d
dice("C:\\Users\\Administrator\\Desktop\\T063935.jpg","C:\\Users\\Administrator\\Desktop\\T063934.jpg" )