在做深度學習的目標檢測中，我們在測試好資料集後，可以用批量檢查兩個資料檔案下面的標籤檔案，看看他們的行數是否一致，這樣可以判斷是否有漏檢的情況，就不用一張張圖片去看了。

對比兩個標籤檔案的行數，可以發現有沒有漏檢。（對比的是打標的和訓練後的標籤）

程式碼如下：

#coding=utf-8
import os,glob
import numpy as np

path1 = 'temp//1//'
path2 = 'temp//2//'

file_one_list = glob.glob(path1 + "*.txt")
file_two_list = glob.glob(path2 + "*.txt")
print(len(file_one_list))

for item in (file_one_list):
 base_name = os.path.basename(item)
 with open(item,'r') as f:
  lines_one=f.readlines()
 cur_file_path = os.path.join(path2,base_name)
 if os.path.exists(cur_file_path):
  with open(cur_file_path,'r') as f:
   lines_two=f.readlines()
  if len(lines_one)!=len(lines_two):
   print("檢測框不對應:",base_name)
 else:
  print("不在在檔案",base_name)

總結

到此這篇關於python批量檢查兩個對應的txt檔案的行數是否一致的文章就介紹到這了,更多相關python批量檢查txt檔案內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們！