def writeOF(): root = "../videos" path = os.path.join(root, "train") startTime = datetime.now() w = 224 h = 224 c = 0 data = {} for path, subdirs, files in os.walk(root): for filename in files: c += 1 count = ofp.writeOpticalFlow(path, filename, w, h, c) data[filename] = count root = "../videos" path = os.path.join(root, "test") w = 224 h = 224 for path, subdirs, files in os.walk(root): for filename in files: c += 1 count = ofp.writeOpticalFlow(path, filename, w, h, c) data[filename] = count with open('../dataset/frame_count.pickle', 'wb') as f: pickle.dump(data, f) print 'Time taken for ' + str(c) + 'testing videos: ' + str( datetime.now() - startTime)
def writeOF(): root = "../videos" path = os.path.join(root, "train") startTime = datetime.now() w=224 h=224 c=0 data={} for path, subdirs, files in os.walk(root): for filename in files: c+=1 count=ofp.writeOpticalFlow(path,filename,w,h,c) data[filename]=count root = "../videos" path = os.path.join(root, "test") w=224 h=224 for path, subdirs, files in os.walk(root): for filename in files: c+=1 count=ofp.writeOpticalFlow(path,filename,w,h,c) data[filename]=count with open('../dataset/frame_count.pickle','wb') as f: pickle.dump(data,f) print 'Time taken for '+str(c)+'testing videos: '+str(datetime.now() - startTime)
def writeOF(): root = "../videos" w = 224 h = 224 c = 0 data = {} for path, subdsirs, files in scandir.walk(root): for filename in files: count = ofp.writeOpticalFlow(path, filename, w, h, c) if count: data[filename] = count print filename c += 1 with open("done.txt", "a") as myfile: myfile.write(filename + '-' + str(c) + '\n') with open('../dataset/frame_count.pickle', 'wb') as f: pickle.dump(data, f)
def writeOF(): root = "../videos" w=224 h=224 c=0 data={} for path, subdsirs, files in scandir.walk(root): for filename in files: count=ofp.writeOpticalFlow(path,filename,w,h,c) if count: data[filename]=count print filename c+=1 with open("done.txt", "a") as myfile: myfile.write(filename+'-'+str(c)+'\n') with open('../dataset/frame_count.pickle','wb') as f: pickle.dump(data,f)
def writeOF(): root = "F:\Dataset/UCF-101" w = 224 h = 224 c = 0 data = {} for path, subdsirs, files in scandir.walk(root): for filename in files: count = ofp.writeOpticalFlow(path, filename, w, h, c) if count: data[filename] = count print(filename) c += 1 with open("done.txt", "a") as myfile: myfile.write(filename + '-' + str(c) + '\n') with open( 'D:\MyDrivers\software/anaconda\wokspace\Video-Classification-2-Stream-CNN-master\dataset/frame_count.pickle', 'wb') as f: pickle.dump(data, f)
def writeOF(): '''create OpticalFlow images ''' root = '../dataset/train' w = 224 h = 224 c = 0 data = {} for path, dirs, files in os.walk(root): for filename in files: filename = filename[:-4] + '.h264' print(filename) count = ofp.writeOpticalFlow(path, filename, w, h, c) if count: data[filename] = count c += 1 with open("done.txt", "a") as myfile: myfile.write(filename + '_' + str(c) + '\n') with open('../dataset/frame_count.pickle', 'wb') as f: pickle.dump(data, f)
def writeOF(): ##root = "../videos" root = '/media/yanrui/文档/迅雷下载/Two-Stream/hmdb51_org/video/' w = 224 h = 224 c = 0 data = {} for path, subdsirs, files in scandir.walk(root): for filename in files: count = ofp.writeOpticalFlow(path, filename, w, h, c) filename = filename.split('.')[0] if count: data[filename] = count print filename c += 1 with open("done.txt", "a") as myfile: myfile.write(filename + '-' + str(c) + '\n') with open('../dataset/frame_count.pickle', 'wb') as f: pickle.dump(data, f)
def main(): if len(sys.argv) > 1 and sys.argv[1] == '--test': pickleFile = "../dataset/temporal_test_data.pickle" fileList = "../dataset/ucfTrainTestlist/testlist01.txt" print "Saving data at: ", pickleFile else: pickleFile = "../dataset/temporal_train_data.pickle" fileList = "../dataset/ucfTrainTestlist/trainlist01.txt" print "Saving data at: ", pickleFile training_data = {} types = CreateDict("../dataset/ucfTrainTestlist/classInd.txt") rootDir = '../dataset/ucf101/' f = open(fileList, 'r') for line in f: filePath = line.split()[0] folder, file = filePath.split('/') count = writeOpticalFlow(rootDir + folder, file, 150, 150, 1) print count blockno = int(count / 50) key = file + '@' + str(blockno) training_data[key] = types[folder] print training_data with open(pickleFile, 'wb') as f: pickle.dump(training_data, f)