8. Salient计数统计代码

import os

count0=0
count1=0
count2=0
count3=0
count4=0
count5=0
count6=0
count7=0
count8=0

create_file = open('SOC_Subitizing.txt', 'w')

pic_names = [i[:-4] for i in os.listdir('./gt/') if 'png' in i]

for pic_name in pic_names:

    if (os.path.exists('./Instance name/' + pic_name + '.txt') == True):
        file = open('./Instance name/' + pic_name + '.txt', 'r')
        line_number = file.readlines()
        create_file.write(str(pic_name) + ',' + str(len(line_number)) + '\n')
        if len(line_number) == 1:
            count1 +=1
        elif len(line_number) == 2:
            count2 += 1
        elif len(line_number) == 3:
            count3 += 1
        elif len(line_number) == 4:
            count4 += 1
        elif len(line_number) == 5:
            count5 += 1
        elif len(line_number) == 6:
            count6 += 1
        elif len(line_number) == 7:
            count7 += 1
        elif len(line_number) == 8:
            count8 += 1
    else:
        create_file.write(str(pic_name) + ',0' + '\n')
        count0 +=1

print(count0)
print(count1)
print(count2)
print(count3)
print(count4)
print(count5)
print(count6)
print(count7)
print(count8)
print(count0+count1+count2+count3+count4+count5+count6+count7+count8)