[Dataset] class instance number checker

class변수의 개수를 추가하여서 실행

#!/usr/bin/env python3
import os
import shutil

dir_path = '/home/jay/DataSets/data/train'

class0 = int()
class1 = int()
class2 = int()
class3 = int()
class4 = int()
class5 = int()
unknown = int()
cnt = int()

breaker1 = False
breaker2 = False

for (root, directories, files) in os.walk(dir_path):
    for file in files:
        if '.txt' in file:
            file_path = os.path.join(root, file)
            with open(file_path, "r") as f:
                lines = f.readlines()                  
                for line in lines:
                    class_num = line.split()[0] 
                    #print("before : ",line)
                    tmp = line.split()
                    if class_num == '0':
                        #tmp[0] = '2'
                        class0 += 1 
                    elif class_num == '1':
                        #tmp[0] = '0'
                        class1 += 1
                    elif class_num == '2':
                        #tmp[0] = '5'
                        class2 += 1
                    elif class_num == '3':
                        #tmp[0] = '5'
                        class3 += 1
                    elif class_num == '4':
                        #tmp[0] = '5'
                        class4 += 1
                    elif class_num == '5':
                        #tmp[0] = '5'
                        class5 += 1
                    else:
                        unknown += 1

print("person : ", class0)
print("bicycle : ", class1)
print("car : ", class2)
print("motocycle : ", class3)
print("bus : ", class4)
print("truck : ", class5)
print("else : ",unknown)