第一个小任务,对I3D网络提取出的特征进行统计(代码待优化)

#-*- coding:utf-8 -*-
import pickle
import os
def get_files(path, rule=".pkl"):
    BTVX = []
    TV1 = []
    TV4 = []
    TV5 = []
    DF = []
    DI = []
    DN = []
    GD = []
    HN = []
    SC = []
    SX = []
    SZ = []
    ZJ = []
    for fpathe, dirs, fs in os.walk(path):  # os.walk是获取所有的目录
        for f in fs:
            if f.startswith('BT'):
                filenameBT = os.path.join(fpathe, f)
                if filenameBT.endswith(rule):  # 判断是否是".pkl"结尾
                    BTVX.append(filenameBT)
            elif f.startswith('CCTV1'):
                filenameTV1 = os.path.join(fpathe, f)
                if filenameTV1.endswith(rule):  # 判断是否是".pkl"结尾
                    TV1.append(filenameTV1)
            elif f.startswith('CCTV4'):
                filenameTV4 = os.path.join(fpathe, f)
                if filenameTV4.endswith(rule):  # 判断是否是".pkl"结尾
                    TV4.append(filenameTV4)
            elif f.startswith('CCTV5'):
                filenameTV5 = os.path.join(fpathe, f)
                if filenameTV5.endswith(rule):  # 判断是否是".pkl"结尾
                    TV5.append(filenameTV5)
            elif f.startswith('DF'):
                filenameDF = os.path.join(fpathe, f)
                if filenameDF.endswith(rule):  # 判断是否是".pkl"结尾
                    DF.append(filenameDF)
            elif f.startswith('DI'):
                filenameDI = os.path.join(fpathe, f)
                if filenameDI.endswith(rule):  # 判断是否是".pkl"结尾
                    DI.append(filenameDI)
            elif f.startswith('DN'):
                filenameDN = os.path.join(fpathe, f)
                if filenameDN.endswith(rule):  # 判断是否是".pkl"结尾
                    DN.append(filenameDN)
            elif f.startswith('GD'):
                filenameGD = os.path.join(fpathe, f)
                if filenameDN.endswith(rule):  # 判断是否是".pkl"结尾
                    GD.append(filenameGD)
            elif f.startswith('HN'):
                filenameHN = os.path.join(fpathe, f)
                if filenameHN.endswith(rule):  # 判断是否是".pkl"结尾
                    HN.append(filenameHN)
            elif f.startswith('SC'):
                filenameSC = os.path.join(fpathe, f)
                if filenameSC.endswith(rule):  # 判断是否是".pkl"结尾
                    SC.append(filenameSC)
            elif f.startswith('SX'):
                filenameSX = os.path.join(fpathe, f)
                if filenameSX.endswith(rule):  # 判断是否是".pkl"结尾
                    SX.append(filenameSX)
            elif f.startswith('SZ'):
                filenameSZ = os.path.join(fpathe, f)
                if filenameSZ.endswith(rule):  # 判断是否是".pkl"结尾
                    SZ.append(filenameSZ)
            elif f.startswith('ZJ'):
                filenameZJ = os.path.join(fpathe, f)
                if filenameZJ.endswith(rule):  # 判断是否是".pkl"结尾
                    ZJ.append(filenameZJ)
    return BTVX,TV1,TV4,TV5,DF,DI,DN,GD,HN,SC,SX,SZ,ZJ
# cate = ['cate1','cate2','cate4','cate6','cate7','cate9','cate10']
# cate = ','.join(cate)
if __name__ == "__main__":
    BT,V1,V4,V5,DF,DI,DN,GD,HN,SC,SX,SZ,ZJ = get_files('video_feature\Jan_videos_feature\Jan_videos\cate1')
    SumBT = 0
    SumV1 = 0
    SumV4 = 0
    SumV5 = 0
    SumDF = 0
    SumDI = 0
    SumDN = 0
    SumGD = 0
    SumHN = 0
    SumSC = 0
    SumSX = 0
    SumSZ = 0
    SumZJ = 0
    with open("Tongji_cate1.txt", "wb") as ff:
        countBT = 0
        countV1 = 0
        countV4 = 0
        countV5 = 0
        countDF = 0
        countDI = 0
        countDN = 0
        countGD = 0
        countHN = 0
        countSC = 0
        countSX = 0
        countSZ = 0
        countZJ = 0
        for i in BT:
            print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countBT = countBT + 1   #第i个plk的特征值个数
        SumBT = SumBT + countBT
        for i in V1:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countV1 = countV1 + 1   #第i个plk的特征值个数
        SumV1 = SumV1 + countV1
        for i in V4:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countV4 = countV4 + 1   #第i个plk的特征值个数
        SumV4 = SumV4 + countV4
        for i in V5:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countV5 = countV5 + 1   #第i个plk的特征值个数
        SumV5 = SumV5 + countV5
        for i in DF:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countDF = countDF + 1   #第i个plk的特征值个数
        SumDF = SumDF + countDF
        for i in DI:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countDI = countDI + 1   #第i个plk的特征值个数
        SumDI = SumDI + countDI
        for i in DN:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countDN = countDN + 1   #第i个plk的特征值个数
        SumDN = SumDN + countDN
        for i in GD:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countGD = countGD + 1   #第i个plk的特征值个数
        SumGD = SumGD + countGD
        for i in HN:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countHN = countHN + 1   #第i个plk的特征值个数
        SumHN = SumHN + countHN
        for i in SC:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countSC = countSC + 1   #第i个plk的特征值个数
        SumSC = SumSC + countSC
        for i in SX:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countSX = countSX + 1   #第i个plk的特征值个数
        SumSX = SumSX + countSX
        for i in SZ:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countSZ = countSZ + 1   #第i个plk的特征值个数
        SumSZ = SumSZ + countSZ
        for i in ZJ:
            # print i
            f = open(str(i),'rb')
            x = pickle.load(f)
            for key in x:
                countZJ = countZJ + 1   #第i个plk的特征值个数
        SumZJ = SumZJ + countZJ
        ff.write('Cate1:'+'\n')
        ff.write('BTVXWHD:'+str(SumBT)+'\n')
        ff.write('CCTV1:' + str(SumV1) + '\n')
        ff.write('CCTV4:' + str(SumV4) + '\n')
        ff.write('CCTV5:' + str(SumV5) + '\n')
        ff.write('DFWS:' + str(SumDF) + '\n')
        ff.write('DICJ:' + str(SumDI) + '\n')
        ff.write('DNWS:' + str(SumDN) + '\n')
        ff.write('GDXW:' + str(SumGD) + '\n')
        ff.write('HNWS:' + str(SumHN) + '\n')
        ff.write('SCWS:' + str(SumSC) + '\n')
        ff.write('SXDSQC:' + str(SumSX) + '\n')
        ff.write('SZWS:' + str(SumSZ) + '\n')
        ff.write('ZJWS:' + str(SumZJ) + '\n')
        # print SumBT
    BT, V1, V4, V5, DF, DI, DN, GD, HN, SC, SX, SZ, ZJ = get_files(
        'video_feature\Jan_videos_feature\Jan_videos\cate2')
    SumBT = 0
    SumV1 = 0
    SumV4 = 0
    SumV5 = 0
    SumDF = 0
    SumDI = 0
    SumDN = 0
    SumGD = 0
    SumHN = 0
    SumSC = 0
    SumSX = 0
    SumSZ = 0
    SumZJ = 0
    with open("Tongji_cate2.txt", "wb") as ff:
        countBT = 0
        countV1 = 0
        countV4 = 0
        countV5 = 0
        countDF = 0
        countDI = 0
        countDN = 0
        countGD = 0
        countHN = 0
        countSC = 0
        countSX = 0
        countSZ = 0
        countZJ = 0
        for i in BT:
            print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countBT = countBT + 1  # 第i个plk的特征值个数
        SumBT = SumBT + countBT
        for i in V1:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV1 = countV1 + 1  # 第i个plk的特征值个数
        SumV1 = SumV1 + countV1
        for i in V4:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV4 = countV4 + 1  # 第i个plk的特征值个数
        SumV4 = SumV4 + countV4
        for i in V5:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV5 = countV5 + 1  # 第i个plk的特征值个数
        SumV5 = SumV5 + countV5
        for i in DF:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDF = countDF + 1  # 第i个plk的特征值个数
        SumDF = SumDF + countDF
        for i in DI:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDI = countDI + 1  # 第i个plk的特征值个数
        SumDI = SumDI + countDI
        for i in DN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDN = countDN + 1  # 第i个plk的特征值个数
        SumDN = SumDN + countDN
        for i in GD:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countGD = countGD + 1  # 第i个plk的特征值个数
        SumGD = SumGD + countGD
        for i in HN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countHN = countHN + 1  # 第i个plk的特征值个数
        SumHN = SumHN + countHN
        for i in SC:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSC = countSC + 1  # 第i个plk的特征值个数
        SumSC = SumSC + countSC
        for i in SX:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSX = countSX + 1  # 第i个plk的特征值个数
        SumSX = SumSX + countSX
        for i in SZ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSZ = countSZ + 1  # 第i个plk的特征值个数
        SumSZ = SumSZ + countSZ
        for i in ZJ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countZJ = countZJ + 1  # 第i个plk的特征值个数
        SumZJ = SumZJ + countZJ
        ff.write('Cate2:' + '\n')
        ff.write('BTVXWHD:' + str(SumBT) + '\n')
        ff.write('CCTV1:' + str(SumV1) + '\n')
        ff.write('CCTV4:' + str(SumV4) + '\n')
        ff.write('CCTV5:' + str(SumV5) + '\n')
        ff.write('DFWS:' + str(SumDF) + '\n')
        ff.write('DICJ:' + str(SumDI) + '\n')
        ff.write('DNWS:' + str(SumDN) + '\n')
        ff.write('GDXW:' + str(SumGD) + '\n')
        ff.write('HNWS:' + str(SumHN) + '\n')
        ff.write('SCWS:' + str(SumSC) + '\n')
        ff.write('SXDSQC:' + str(SumSX) + '\n')
        ff.write('SZWS:' + str(SumSZ) + '\n')
        ff.write('ZJWS:' + str(SumZJ) + '\n')
        # print i
    BT, V1, V4, V5, DF, DI, DN, GD, HN, SC, SX, SZ, ZJ = get_files(
        'video_feature\Jan_videos_feature\Jan_videos\cate4')
    SumBT = 0
    SumV1 = 0
    SumV4 = 0
    SumV5 = 0
    SumDF = 0
    SumDI = 0
    SumDN = 0
    SumGD = 0
    SumHN = 0
    SumSC = 0
    SumSX = 0
    SumSZ = 0
    SumZJ = 0
    with open("Tongji_cate4.txt", "wb") as ff:
        countBT = 0
        countV1 = 0
        countV4 = 0
        countV5 = 0
        countDF = 0
        countDI = 0
        countDN = 0
        countGD = 0
        countHN = 0
        countSC = 0
        countSX = 0
        countSZ = 0
        countZJ = 0
        for i in BT:
            print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countBT = countBT + 1  # 第i个plk的特征值个数
        SumBT = SumBT + countBT
        for i in V1:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV1 = countV1 + 1  # 第i个plk的特征值个数
        SumV1 = SumV1 + countV1
        for i in V4:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV4 = countV4 + 1  # 第i个plk的特征值个数
        SumV4 = SumV4 + countV4
        for i in V5:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV5 = countV5 + 1  # 第i个plk的特征值个数
        SumV5 = SumV5 + countV5
        for i in DF:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDF = countDF + 1  # 第i个plk的特征值个数
        SumDF = SumDF + countDF
        for i in DI:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDI = countDI + 1  # 第i个plk的特征值个数
        SumDI = SumDI + countDI
        for i in DN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDN = countDN + 1  # 第i个plk的特征值个数
        SumDN = SumDN + countDN
        for i in GD:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countGD = countGD + 1  # 第i个plk的特征值个数
        SumGD = SumGD + countGD
        for i in HN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countHN = countHN + 1  # 第i个plk的特征值个数
        SumHN = SumHN + countHN
        for i in SC:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSC = countSC + 1  # 第i个plk的特征值个数
        SumSC = SumSC + countSC
        for i in SX:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSX = countSX + 1  # 第i个plk的特征值个数
        SumSX = SumSX + countSX
        for i in SZ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSZ = countSZ + 1  # 第i个plk的特征值个数
        SumSZ = SumSZ + countSZ
        for i in ZJ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countZJ = countZJ + 1  # 第i个plk的特征值个数
        SumZJ = SumZJ + countZJ
        ff.write('Cate4:' + '\n')
        ff.write('BTVXWHD:' + str(SumBT) + '\n')
        ff.write('CCTV1:' + str(SumV1) + '\n')
        ff.write('CCTV4:' + str(SumV4) + '\n')
        ff.write('CCTV5:' + str(SumV5) + '\n')
        ff.write('DFWS:' + str(SumDF) + '\n')
        ff.write('DICJ:' + str(SumDI) + '\n')
        ff.write('DNWS:' + str(SumDN) + '\n')
        ff.write('GDXW:' + str(SumGD) + '\n')
        ff.write('HNWS:' + str(SumHN) + '\n')
        ff.write('SCWS:' + str(SumSC) + '\n')
        ff.write('SXDSQC:' + str(SumSX) + '\n')
        ff.write('SZWS:' + str(SumSZ) + '\n')
        ff.write('ZJWS:' + str(SumZJ) + '\n')

    BT, V1, V4, V5, DF, DI, DN, GD, HN, SC, SX, SZ, ZJ = get_files(
        'video_feature\Jan_videos_feature\Jan_videos\cate6')
    SumBT = 0
    SumV1 = 0
    SumV4 = 0
    SumV5 = 0
    SumDF = 0
    SumDI = 0
    SumDN = 0
    SumGD = 0
    SumHN = 0
    SumSC = 0
    SumSX = 0
    SumSZ = 0
    SumZJ = 0
    with open("Tongji_cate6.txt", "wb") as ff:
        countBT = 0
        countV1 = 0
        countV4 = 0
        countV5 = 0
        countDF = 0
        countDI = 0
        countDN = 0
        countGD = 0
        countHN = 0
        countSC = 0
        countSX = 0
        countSZ = 0
        countZJ = 0
        for i in BT:
            print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countBT = countBT + 1  # 第i个plk的特征值个数
        SumBT = SumBT + countBT
        for i in V1:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV1 = countV1 + 1  # 第i个plk的特征值个数
        SumV1 = SumV1 + countV1
        for i in V4:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV4 = countV4 + 1  # 第i个plk的特征值个数
        SumV4 = SumV4 + countV4
        for i in V5:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV5 = countV5 + 1  # 第i个plk的特征值个数
        SumV5 = SumV5 + countV5
        for i in DF:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDF = countDF + 1  # 第i个plk的特征值个数
        SumDF = SumDF + countDF
        for i in DI:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDI = countDI + 1  # 第i个plk的特征值个数
        SumDI = SumDI + countDI
        for i in DN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDN = countDN + 1  # 第i个plk的特征值个数
        SumDN = SumDN + countDN
        for i in GD:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countGD = countGD + 1  # 第i个plk的特征值个数
        SumGD = SumGD + countGD
        for i in HN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countHN = countHN + 1  # 第i个plk的特征值个数
        SumHN = SumHN + countHN
        for i in SC:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSC = countSC + 1  # 第i个plk的特征值个数
        SumSC = SumSC + countSC
        for i in SX:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSX = countSX + 1  # 第i个plk的特征值个数
        SumSX = SumSX + countSX
        for i in SZ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSZ = countSZ + 1  # 第i个plk的特征值个数
        SumSZ = SumSZ + countSZ
        for i in ZJ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countZJ = countZJ + 1  # 第i个plk的特征值个数
        SumZJ = SumZJ + countZJ
        ff.write('Cate6:' + '\n')
        ff.write('BTVXWHD:' + str(SumBT) + '\n')
        ff.write('CCTV1:' + str(SumV1) + '\n')
        ff.write('CCTV4:' + str(SumV4) + '\n')
        ff.write('CCTV5:' + str(SumV5) + '\n')
        ff.write('DFWS:' + str(SumDF) + '\n')
        ff.write('DICJ:' + str(SumDI) + '\n')
        ff.write('DNWS:' + str(SumDN) + '\n')
        ff.write('GDXW:' + str(SumGD) + '\n')
        ff.write('HNWS:' + str(SumHN) + '\n')
        ff.write('SCWS:' + str(SumSC) + '\n')
        ff.write('SXDSQC:' + str(SumSX) + '\n')
        ff.write('SZWS:' + str(SumSZ) + '\n')
        ff.write('ZJWS:' + str(SumZJ) + '\n')

    BT, V1, V4, V5, DF, DI, DN, GD, HN, SC, SX, SZ, ZJ = get_files(
        'video_feature\Jan_videos_feature\Jan_videos\cate7')
    SumBT = 0
    SumV1 = 0
    SumV4 = 0
    SumV5 = 0
    SumDF = 0
    SumDI = 0
    SumDN = 0
    SumGD = 0
    SumHN = 0
    SumSC = 0
    SumSX = 0
    SumSZ = 0
    SumZJ = 0
    with open("Tongji_cate7.txt", "wb") as ff:
        countBT = 0
        countV1 = 0
        countV4 = 0
        countV5 = 0
        countDF = 0
        countDI = 0
        countDN = 0
        countGD = 0
        countHN = 0
        countSC = 0
        countSX = 0
        countSZ = 0
        countZJ = 0
        for i in BT:
            print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countBT = countBT + 1  # 第i个plk的特征值个数
        SumBT = SumBT + countBT
        for i in V1:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV1 = countV1 + 1  # 第i个plk的特征值个数
        SumV1 = SumV1 + countV1
        for i in V4:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV4 = countV4 + 1  # 第i个plk的特征值个数
        SumV4 = SumV4 + countV4
        for i in V5:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV5 = countV5 + 1  # 第i个plk的特征值个数
        SumV5 = SumV5 + countV5
        for i in DF:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDF = countDF + 1  # 第i个plk的特征值个数
        SumDF = SumDF + countDF
        for i in DI:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDI = countDI + 1  # 第i个plk的特征值个数
        SumDI = SumDI + countDI
        for i in DN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDN = countDN + 1  # 第i个plk的特征值个数
        SumDN = SumDN + countDN
        for i in GD:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countGD = countGD + 1  # 第i个plk的特征值个数
        SumGD = SumGD + countGD
        for i in HN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countHN = countHN + 1  # 第i个plk的特征值个数
        SumHN = SumHN + countHN
        for i in SC:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSC = countSC + 1  # 第i个plk的特征值个数
        SumSC = SumSC + countSC
        for i in SX:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSX = countSX + 1  # 第i个plk的特征值个数
        SumSX = SumSX + countSX
        for i in SZ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSZ = countSZ + 1  # 第i个plk的特征值个数
        SumSZ = SumSZ + countSZ
        for i in ZJ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countZJ = countZJ + 1  # 第i个plk的特征值个数
        SumZJ = SumZJ + countZJ
        ff.write('Cate7:' + '\n')
        ff.write('BTVXWHD:' + str(SumBT) + '\n')
        ff.write('CCTV1:' + str(SumV1) + '\n')
        ff.write('CCTV4:' + str(SumV4) + '\n')
        ff.write('CCTV5:' + str(SumV5) + '\n')
        ff.write('DFWS:' + str(SumDF) + '\n')
        ff.write('DICJ:' + str(SumDI) + '\n')
        ff.write('DNWS:' + str(SumDN) + '\n')
        ff.write('GDXW:' + str(SumGD) + '\n')
        ff.write('HNWS:' + str(SumHN) + '\n')
        ff.write('SCWS:' + str(SumSC) + '\n')
        ff.write('SXDSQC:' + str(SumSX) + '\n')
        ff.write('SZWS:' + str(SumSZ) + '\n')
        ff.write('ZJWS:' + str(SumZJ) + '\n')

    BT, V1, V4, V5, DF, DI, DN, GD, HN, SC, SX, SZ, ZJ = get_files(
        'video_feature\Jan_videos_feature\Jan_videos\cate9')
    SumBT = 0
    SumV1 = 0
    SumV4 = 0
    SumV5 = 0
    SumDF = 0
    SumDI = 0
    SumDN = 0
    SumGD = 0
    SumHN = 0
    SumSC = 0
    SumSX = 0
    SumSZ = 0
    SumZJ = 0
    with open("Tongji_cate9.txt", "wb") as ff:
        countBT = 0
        countV1 = 0
        countV4 = 0
        countV5 = 0
        countDF = 0
        countDI = 0
        countDN = 0
        countGD = 0
        countHN = 0
        countSC = 0
        countSX = 0
        countSZ = 0
        countZJ = 0
        for i in BT:
            print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countBT = countBT + 1  # 第i个plk的特征值个数
        SumBT = SumBT + countBT
        for i in V1:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV1 = countV1 + 1  # 第i个plk的特征值个数
        SumV1 = SumV1 + countV1
        for i in V4:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV4 = countV4 + 1  # 第i个plk的特征值个数
        SumV4 = SumV4 + countV4
        for i in V5:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV5 = countV5 + 1  # 第i个plk的特征值个数
        SumV5 = SumV5 + countV5
        for i in DF:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDF = countDF + 1  # 第i个plk的特征值个数
        SumDF = SumDF + countDF
        for i in DI:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDI = countDI + 1  # 第i个plk的特征值个数
        SumDI = SumDI + countDI
        for i in DN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDN = countDN + 1  # 第i个plk的特征值个数
        SumDN = SumDN + countDN
        for i in GD:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countGD = countGD + 1  # 第i个plk的特征值个数
        SumGD = SumGD + countGD
        for i in HN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countHN = countHN + 1  # 第i个plk的特征值个数
        SumHN = SumHN + countHN
        for i in SC:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSC = countSC + 1  # 第i个plk的特征值个数
        SumSC = SumSC + countSC
        for i in SX:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSX = countSX + 1  # 第i个plk的特征值个数
        SumSX = SumSX + countSX
        for i in SZ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSZ = countSZ + 1  # 第i个plk的特征值个数
        SumSZ = SumSZ + countSZ
        for i in ZJ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countZJ = countZJ + 1  # 第i个plk的特征值个数
        SumZJ = SumZJ + countZJ
        ff.write('Cate9:' + '\n')
        ff.write('BTVXWHD:' + str(SumBT) + '\n')
        ff.write('CCTV1:' + str(SumV1) + '\n')
        ff.write('CCTV4:' + str(SumV4) + '\n')
        ff.write('CCTV5:' + str(SumV5) + '\n')
        ff.write('DFWS:' + str(SumDF) + '\n')
        ff.write('DICJ:' + str(SumDI) + '\n')
        ff.write('DNWS:' + str(SumDN) + '\n')
        ff.write('GDXW:' + str(SumGD) + '\n')
        ff.write('HNWS:' + str(SumHN) + '\n')
        ff.write('SCWS:' + str(SumSC) + '\n')
        ff.write('SXDSQC:' + str(SumSX) + '\n')
        ff.write('SZWS:' + str(SumSZ) + '\n')
        ff.write('ZJWS:' + str(SumZJ) + '\n')

    BT, V1, V4, V5, DF, DI, DN, GD, HN, SC, SX, SZ, ZJ = get_files(
        'video_feature\Jan_videos_feature\Jan_videos\cate10')
    SumBT = 0
    SumV1 = 0
    SumV4 = 0
    SumV5 = 0
    SumDF = 0
    SumDI = 0
    SumDN = 0
    SumGD = 0
    SumHN = 0
    SumSC = 0
    SumSX = 0
    SumSZ = 0
    SumZJ = 0
    with open("Tongji_cate10.txt", "wb") as ff:
        countBT = 0
        countV1 = 0
        countV4 = 0
        countV5 = 0
        countDF = 0
        countDI = 0
        countDN = 0
        countGD = 0
        countHN = 0
        countSC = 0
        countSX = 0
        countSZ = 0
        countZJ = 0
        for i in BT:
            print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countBT = countBT + 1  # 第i个plk的特征值个数
        SumBT = SumBT + countBT
        for i in V1:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV1 = countV1 + 1  # 第i个plk的特征值个数
        SumV1 = SumV1 + countV1
        for i in V4:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV4 = countV4 + 1  # 第i个plk的特征值个数
        SumV4 = SumV4 + countV4
        for i in V5:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countV5 = countV5 + 1  # 第i个plk的特征值个数
        SumV5 = SumV5 + countV5
        for i in DF:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDF = countDF + 1  # 第i个plk的特征值个数
        SumDF = SumDF + countDF
        for i in DI:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDI = countDI + 1  # 第i个plk的特征值个数
        SumDI = SumDI + countDI
        for i in DN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countDN = countDN + 1  # 第i个plk的特征值个数
        SumDN = SumDN + countDN
        for i in GD:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countGD = countGD + 1  # 第i个plk的特征值个数
        SumGD = SumGD + countGD
        for i in HN:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countHN = countHN + 1  # 第i个plk的特征值个数
        SumHN = SumHN + countHN
        for i in SC:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSC = countSC + 1  # 第i个plk的特征值个数
        SumSC = SumSC + countSC
        for i in SX:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSX = countSX + 1  # 第i个plk的特征值个数
        SumSX = SumSX + countSX
        for i in SZ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countSZ = countSZ + 1  # 第i个plk的特征值个数
        SumSZ = SumSZ + countSZ
        for i in ZJ:
            # print i
            f = open(str(i), 'rb')
            x = pickle.load(f)
            for key in x:
                countZJ = countZJ + 1  # 第i个plk的特征值个数
        SumZJ = SumZJ + countZJ
        ff.write('Cate10:' + '\n')
        ff.write('BTVXWHD:' + str(SumBT) + '\n')
        ff.write('CCTV1:' + str(SumV1) + '\n')
        ff.write('CCTV4:' + str(SumV4) + '\n')
        ff.write('CCTV5:' + str(SumV5) + '\n')
        ff.write('DFWS:' + str(SumDF) + '\n')
        ff.write('DICJ:' + str(SumDI) + '\n')
        ff.write('DNWS:' + str(SumDN) + '\n')
        ff.write('GDXW:' + str(SumGD) + '\n')
        ff.write('HNWS:' + str(SumHN) + '\n')
        ff.write('SCWS:' + str(SumSC) + '\n')
        ff.write('SXDSQC:' + str(SumSX) + '\n')
        ff.write('SZWS:' + str(SumSZ) + '\n')
        ff.write('ZJWS:' + str(SumZJ) + '\n')

猜你喜欢

转载自blog.csdn.net/mr_brooks/article/details/80199504