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
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
if len(img.shape) == 3:
img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
subStartTime = time.time()
first_vertical, fire_lane_map_vertical, count_standalone_vertical = Dmtler.split(
img, 0)
first_horizontal, fire_lane_map_horizontal, count_standalone_horizontal = Dmtler.split(
img, 1)
spendTime = time.time()
countSplitTime += spendTime - subStartTime
subStartTime = spendTime
print countSplitTime
# print count_standalone_vertical, count_standalone_horizontal
first_vertical = Dmtler.merge(img, first_vertical)
first_horizontal = Dmtler.merge(img, first_horizontal)
spendTime = time.time()
countMergeTime += time.time() - subStartTime
subStartTime = spendTime
print countMergeTime
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
if len(img.shape) == 3:
img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
subStartTime = time.time()
first_vertical, fire_lane_map_vertical, count_standalone_vertical = Dmtler.split(img, 0)
first_horizontal, fire_lane_map_horizontal, count_standalone_horizontal = Dmtler.split(img, 1)
spendTime = time.time()
countSplitTime += spendTime - subStartTime
subStartTime = spendTime
print countSplitTime
# print count_standalone_vertical, count_standalone_horizontal
first_vertical = Dmtler.merge(img, first_vertical)
first_horizontal = Dmtler.merge(img, first_horizontal)
spendTime = time.time()
countMergeTime += time.time() - subStartTime
subStartTime = spendTime
print countMergeTime
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)
if len(img.shape) == 3:
img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
root0, count_standalone = Dmtler.split(img, 0)
map0 = Dmtler.getSalientMap(img, root0)
root1, count_standalone = Dmtler.split(img, 1)
map1 = Dmtler.getSalientMap(img, root1)
map = (map0 + map1) / 2
# Dismantler.showImg(map)
savePath = os.path.join(resultPath, f)
# print savePath
cv.imwrite(savePath, map)
nImageAll += 1
if nImageAll % 100 == 0:
print '%d images have been mapped.' % nImageAll
