def processFolder(datasetPath, binaryRootPath): """Call your executable for all sequences in all categories.""" stats = Stats(datasetPath) #STATS for category in getDirectories(datasetPath): categoryPath = os.path.join(datasetPath, category) if isValidVideoFolder(categoryPath): stats.addCategories(category) #STATS for video in getDirectories(categoryPath): videoPath = categoryPath if video != "FramesRGB" and video != "FramesT" and video != "groundtruthRgb" and video != "groundtruthT": binaryPath = os.path.join(binaryRootPath, category, video) if isValidResultsFolder(binaryPath): confusionMatrix = compareWithGroungtruth( videoPath, os.path.join(binaryPath, 'MoG')) stats.update(category, video + '/MoG', confusionMatrix) confusionMatrix = compareWithGroungtruth( videoPath, os.path.join(binaryPath, 'SubSENSE')) stats.update(category, video + '/SubSENSE', confusionMatrix) stats.writeCategoryResult(category) stats.writeOverallResults()
def processFolder(datasetPath, binaryRootPath): """Call your executable for all sequences in all categories.""" stats = Stats(datasetPath) #STATS for category in getDirectories(datasetPath): stats.addCategories(category) #STATS categoryPath = os.path.join(datasetPath, category) for video in getDirectories(categoryPath): videoPath = os.path.join(categoryPath, video) binaryPath = os.path.join(binaryRootPath, category, video) if isValidVideoFolder(videoPath): confusionMatrix = compareWithGroungtruth(videoPath, binaryPath) stats.update(category, video, confusionMatrix) stats.writeCategoryResult(category) stats.writeOverallResults()
def processFolder(datasetPath, binaryRootPath): """Call your executable for all sequences in all categories.""" stats = Stats(datasetPath) #STATS f = open(datasetPath + '\\' + 'fscore.txt', 'w') for category in getDirectories(datasetPath): stats.addCategories(category) #STATS categoryPath = os.path.join(datasetPath, category) for video in getDirectories(categoryPath): videoPath = os.path.join(categoryPath, video) binaryPath = os.path.join(binaryRootPath, category, video) if isValidVideoFolder(videoPath): confusionMatrix = compareWithGroungtruth(videoPath, binaryPath) stats.update(category, video, confusionMatrix) alpha = 0.000001 fscore = (2.0 * confusionMatrix[0])/ (((2.0 * confusionMatrix[0]) + confusionMatrix[1] + confusionMatrix[2]) + alpha) f.write(video + ' : ' + str(fscore) + '\n') else: print ('Invalid folder : ' + videoPath) stats.writeCategoryResult(category) stats.writeOverallResults() f.close()