#讀取txt-標(biāo)準(zhǔn)txt為基準(zhǔn)-分類別求閾值-閾值為0. 0.3 0.5 0.7的統(tǒng)計(jì)
import glob
import os
import numpy as np
 
#設(shè)定的閾值
threshold1=0.3
threshold2=0.5
threshold3=0.7
 
#閾值計(jì)數(shù)器
counter0=0
counter1=0
counter2=0
counter3=0
 
stdtxt=''#標(biāo)注txt路徑
testtxt=''#測(cè)試txt路徑
 
txtlist=glob.glob(r'%s\*.txt' %stdtxt)#獲取所有txt文件
for path in txtlist:#對(duì)每個(gè)txt操作
  
  txtname=os.path.basename(path)[:-4]#獲取txt文件名
  label=1
  eachtxt=np.loadtxt(path) #讀取文件
  for line in eachtxt:
    if line[0]==label:
      #構(gòu)建背景為0框?yàn)?的圖
      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=[]#用來(lái)存儲(chǔ)所有iou的集合
      for tline in testfile:#對(duì)測(cè)試txt的每行進(jìn)行操作
        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 i in map3:
            if i==2:
              a=a+1
          iou=a/(line[3]*line[4]+tline[3]*tline[4]-a)#計(jì)算iou
          iou_list.append(iou)#添加到集合尾部
          
      threshold=max(iou_list)#閾值取最大的
      #閾值統(tǒng)計(jì)
      if threshold>=threshold3:
        counter3=counter3+1
      elif threshold>=threshold2:
        counter2=counter2+1
      elif threshold>=threshold1:
        counter1=counter1+1
      elif threshold<threshold1:#漏檢
        counter0=counter0+1