訓練完目標檢測模型之后，需要評價其性能，在不同的閾值下的準確度是多少，有沒有漏檢，在這里基于IoU(Intersection over Union)來計算。

希望能提供一些思路，如果覺得有用歡迎贊我表揚我~

IoU的值可以理解為系統預測出來的框與原來圖片中標記的框的重合程度。系統預測出來的框是利用目標檢測模型對測試數據集進行識別得到的。

計算方法即檢測結果DetectionResult與GroundTruth的交集比上它們的并集，如下圖：

藍色的框是：GroundTruth

黃色的框是：DetectionResult

綠色的框是：DetectionResult ?GroundTruth

紅色的框是：DetectionResult ?GroundTruth

基本思路是先讀取原來圖中標記的框信息，對每一張圖，把所需要的那一個類別的框拿出來，與測試集上識別出來的框進行比較，計算IoU，選擇最大的值作為當前框的IoU值，然后通過設定的閾值（漏檢0, 0.3, 0.5, 0.7）來進行比較統計，最后得到每個閾值下的所有的判定為正確檢測（IoU值大于閾值）的框的數量，然后與原本的標記框的數量一起計算準確度。

其中計算IoU的時候是重新構建一個背景為0的圖，設定框所在的位置為1，分別利用原本標注的框和測試識別的框來構建兩個這樣的圖，兩者相加就能夠讓重疊的部分變成2，于是就可以知道重疊部分的大小（交集），從而計算IoU。

構建代碼如下：

?

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

#讀取txt-標準txt為基準-分類別求閾值-閾值為0. 0.3 0.5 0.7的統計

import glob

import os

import numpy as np

#設定的閾值

threshold1=0.3

threshold2=0.5

threshold3=0.7

#閾值計數器

counter0=0

counter1=0

counter2=0

counter3=0

stdtxt=''#標注txt路徑

testtxt=''#測試txt路徑

txtlist=glob.glob(r'%s\*.txt' %stdtxt)#獲取所有txt文件

for pathin txtlist:#對每個txt操作

txtname=os.path.basename(path)[:-4]#獲取txt文件名

label=1

eachtxt=np.loadtxt(path)#讀取文件

for linein eachtxt:

if line[0]==label:

#構建背景為0框為1的圖

map1=np.zeros((960,1280))

map1[line[2]:(line[2]+line[4]),line[1]:(line[1]+line[3])]=1

testfile=np.loadtxt(testtxt+ txtname+ '.txt')

c=0

iou_list=[]#用來存儲所有iou的集合

for tlinein testfile:#對測試txt的每行進行操作

if tline[0]==label:

c=c+1

map2=np.zeros((960,1280))

map2[tline[2]:(tline[2]+tline[4]),tline[1]:(tline[1]+tline[3])]=1

map3=map1+map2

a=0

for iin map3:

if i==2:

a=a+1

iou=a/(line[3]*line[4]+tline[3]*tline[4]-a)#計算iou

iou_list.append(iou)#添加到集合尾部

threshold=max(iou_list)#閾值取最大的

#閾值統計

if threshold>=threshold3:

counter3=counter3+1

elif threshold>=threshold2:

counter2=counter2+1

elif threshold>=threshold1:

counter1=counter1+1

elif threshold

counter0=counter0+1

以上這篇python:目標檢測模型預測準確度計算方式(基于IoU)就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持服務器之家。

原文鏈接：https://blog.csdn.net/malvas/article/details/84957820

總結

以上是生活随笔為你收集整理的python模型预测_python:目标检测模型预测准确度计算方式(基于IoU)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。