def getFeatureByFireLaneMapFromFileList(CID, fileList, showStatus = True):
Opt_Dmtler = Option_Dismantler(isTrain = False)
Dmtler = Dismantler(Opt_Dmtler)
data = np.zeros([len(fileList), 2 + CID.division[1] * CID.division[0]])
for (i,filename) in enumerate(fileList):
img = cv.imread(filename)
if len(img.shape) == 3:
img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
first_vertical, fire_lane_map_vertical, count_standalone_first_vertical = Dmtler.split(img, 0)
first_horizontal, fire_lane_map_horizontal, count_standalone_first_horizontal = Dmtler.split(img, 1)
map = fire_lane_map_vertical + fire_lane_map_horizontal
# Dmtler.showImg(fire_lane_map_vertical)
# Dmtler.showImg(fire_lane_map_horizontal)
# Dmtler.showImg(map)
data[i, :] = CID.getFeatureByFireLaneMap(map)
if showStatus:
print '%d / %d images has been collected' %(i, len(fileList))
print '%d images has been collected.' %len(fileList)
return data
import time
def getModelPath(path, dirName):
dirName = dirName + datetime.datetime.now().strftime("%Y-%m-%d")
return Common.makeDir(path, dirName)
# corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/Corpus/Classifier/VizSet_pm_ee_cat0124_pure/scheme'
corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/tester'
# corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/Corpus/Dismantler/test_corpus/large_corpus'
resultPath = '/Users/sephon/Desktop/Research/VizioMetrics/Dismantler/Result'
resultPath = getModelPath(resultPath, '')
Opt_dmtler = Option_Dismantler(isTrain=False)
Dmtler = Dismantler(Opt_dmtler)
nImageAll = 0
startTime = time.time()
countSingle = 0
countMulti = 0
countSplitTime = 0
countMergeTime = 0
countSelectTime = 0
for dirPath, dirNames, fileNames in os.walk(corpusPath):
for f in fileNames:
fname, suffix = Common.getFileNameAndSuffix(f)
if suffix in Opt_dmtler.validImageFormat:
import cv2 as cv
import os
import time
def getModelPath(path, dirName):
dirName = dirName + datetime.datetime.now().strftime("%Y-%m-%d")
return Common.makeDir(path, dirName)
# corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/Corpus/Classifier/VizSet_pm_ee_cat0124_pure/scheme'
corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/tester'
# corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/Corpus/Dismantler/test_corpus/large_corpus'
resultPath = '/Users/sephon/Desktop/Research/VizioMetrics/Dismantler/Result'
resultPath = getModelPath(resultPath, '')
Opt_dmtler = Option_Dismantler(isTrain = False)
Dmtler = Dismantler(Opt_dmtler)
nImageAll = 0
startTime = time.time()
countSingle = 0
countMulti = 0
countSplitTime = 0
countMergeTime = 0
countSelectTime = 0
for dirPath, dirNames, fileNames in os.walk(corpusPath):
for f in fileNames:
fname, suffix = Common.getFileNameAndSuffix(f)
if suffix in Opt_dmtler.validImageFormat:
import os
import time
def getModelPath(path, dirName):
dirName = dirName + datetime.datetime.now().strftime("%Y-%m-%d")
return Common.makeDir(path, dirName)
# corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/Corpus/Dismantler/test_corpus/large_corpus'
# corpusPath = '/Users/sephon/Desktop/Research/VizioMetrics/Corpus/Classifier/VizSet_pm_ee_cat0124_pure/equation'
corpusPath = '/Users/sephon/Desktop/Research/ReVision/corpus/pm_ee_classified/cat_0/mix'
resultPath = '/Users/sephon/Desktop/Research/VizioMetrics/Dismantler/Result'
resultPath = getModelPath(resultPath, '')
Opt_dmtler = Option_Dismantler(isTrain = False)
Dmtler = Dismantler(Opt_dmtler)
nImageAll = 0
startTime = time.time()
for dirPath, dirNames, fileNames in os.walk(corpusPath):
for f in fileNames:
fname, suffix = Common.getFileNameAndSuffix(f)
if suffix in Opt_dmtler.validImageFormat:
filename = os.path.join(dirPath, f)
# Loading Images
img = cv.imread(filename)
print filename
# node_list = Dmtler.dismantle(img)
