读取文件夹下所有的.txt文件;计算两种缺陷在原图的大小(宽,高)
import os
sumw_0 = []
sumh_0 = []
sumw_1 = []
sumh_1 = []
pathDir = os.listdir(r"D:\learn\lianxi\python_cnn\imgs")
for s in pathDir:
newDir = os.path.join(r"D:\learn\lianxi\python_cnn\imgs", s)
if os.path.isfile(newDir):
if os.path.splitext(newDir)[1] == ".txt":
file = open(newDir, 'r+')
for line in file.readlines():
line = line.strip() # 去掉每行头尾空白
line_split = line.split()
# print(line_split[3:])
int_line = [float(x) for x in line_split]
# print(int_line)
if int_line[0] == 0.0:
sumw_0.append(int_line[3])
sumh_0.append(int_line[4])
if int_line[0] == 1.0:
sumw_1.append(int_line[3])
sumh_1.append(int_line[4])
meanw_0 = sum(sumw_0) / len(sumw_0) * 416
meanh_0 = sum(sumh_0) / len(sumh_0) * 416
meanw_1 = sum(sumw_1) / len(sumw_1) * 416
meanh_1 = sum(sumh_1) / len(sumh_1) * 416
print("缺陷0的宽度的平均值:%f ; \n缺陷0的高度的平均值:%f ;\n缺陷1的宽度的平均值:%f ; \n缺陷1的高度的平均值:%f ;" % (meanw_0, meanh_0,meanw_1,meanh_1))
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190424160337825.PNG?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NDgwNDUzNg==,size_16,color_FFFFFF,t_70)