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)