计算物体数量的程序（python_pang,临时保存)

import os
import glob

cls_path = '/home/wgb/Desktop/SSD/SSD1011_pang/pang_dataset/cls/*.txt'
file_paths = glob.glob(cls_path)

dice_num = 0
screw_num = 0
screw_driver_number = 0
lego_number = 0
pen_number = 0
tape_number = 0

conts = []
all_number = 0
for file_path in file_paths:
    conts = []
    with open(file_path,'r') as f:
        cont = f.readline()
        while cont:
            conts.append(cont)
            cont = f.readline()

    for i in range(1,len(conts)):
        all_number+=1
        object_name = conts[i].split(',')[1].split(':')[0]

        if object_name=='dice':
            dice_num +=1
        elif object_name=='screw':
            screw_num +=1
        elif object_name == 'lego':
            lego_number += 1
        elif object_name == 'pen':
            pen_number += 1
        elif object_name == 'tape':
            tape_number += 1
        elif object_name == 'screwdriver':
            screw_driver_number += 1

print('screw_driver_number:',screw_driver_number)
print('dice_num:',dice_num)
print('lego_number:',lego_number)
print('pen_number:',pen_number)
print('tape_number:',tape_number)
print('screw_num:',screw_num)
print('all_number:',all_number)