def createFeatureFileGrayScaleSequences(imageSize, imageDirectory, featuresDirectory, featuresName, log):
directory = imageDirectory
classesPath = os.listdir(directory)
featuresSetGray = []
classNumber = 0
for classs in classesPath:
sequences = os.listdir(directory+"/"+classs)
for s in sequences:
files = os.listdir(directory+os.sep+classs+os.sep+s+os.sep)
for image in files:
if (not "txt" in image and not "db" in image):
log.printMessage(("Reading:", directory+os.sep+classs+os.sep+s+os.sep+image))
img = cv2.imread(directory+os.sep+classs+os.sep+s+os.sep+image)
featuresGray = grayImage(img,imageSize,False,"")
featuresGray = whiten(featuresGray)
featuresGray = resize(featuresGray,imageSize)
newFeatures = []
newFeatures.append(int(classNumber))
for x in featuresGray:
for y in x:
newFeatures.append(y)
featuresSetGray.append(newFeatures)
classNumber = classNumber+1
DataUtil.writeSingleFile(featuresSetGray,featuresDirectory+os.sep+featuresName, False)
def createFeatureFile(imageSize, imageDirectory, featuresDirectory, featuresName, log, inputType):
print "Input type:", inputType, "-", DataUtil.INPUT_TYPE["Sequence"]
if inputType == DataUtil.INPUT_TYPE["3D"] or inputType == DataUtil.INPUT_TYPE["Sequence"]:
createFeatureFileGrayScaleSequences(imageSize, imageDirectory,featuresDirectory, featuresName,log)
else:
directory = imageDirectory
classesPath = os.listdir(directory)
featuresSetGray = []
classNumber = 0
for classs in classesPath:
files = os.listdir(directory+os.sep+classs+os.sep)
for image in files:
if ( not "db" in image and not ".directory" in image):
log.printMessage(("Reading:", directory+os.sep+classs+os.sep+image))
img = Image.open(directory+os.sep+classs+os.sep+image)
img = numpy.array(img)
if len(img.shape) == 3:
b,g,r = cv2.split(img)
img = cv2.merge([r,g,b])
#cv2.imwrite("/informatik2/wtm/home/barros/demo_ws/src/dialog/scripts/networkPosture/test/test1.jpg",img)
if not inputType == DataUtil.INPUT_TYPE["Color"]:
img = grayImage(img,imageSize,False,"")
img = whiten(img)
#cv2.imwrite("/informatik2/wtm/home/barros/demo_ws/src/dialog/scripts/networkPosture/test/test2.jpg",img)
img = resize(img,imageSize)
#cv2.imwrite("/informatik2/wtm/home/barros/demo_ws/src/dialog/scripts/networkPosture/test/test3.jpg",img)
newFeatures = []
newFeatures.append(int(classNumber))
for x in img:
for y in x:
if inputType == DataUtil.INPUT_TYPE["Color"]:
colors = []
for c in y:
colors.append(c)
newFeatures.append(colors)
else:
newFeatures.append(y)
featuresSetGray.append(newFeatures)
classNumber = classNumber+1
#print "Total features:", len(featuresSetGray)
DataUtil.writeSingleFile(featuresSetGray,featuresDirectory+os.sep+featuresName, inputType == DataUtil.INPUT_TYPE["Color"])
def createFeatureFile(imageSize, imageDirectory, featuresDirectory, featuresName, log, inputType):
if inputType == DataUtil.INPUT_TYPE["3D"]:
createFeatureFileGrayScaleSequences(imageSize, imageDirectory,featuresDirectory, featuresName,log)
else:
directory = imageDirectory
classesPath = os.listdir(directory)
featuresSetGray = []
classNumber = 0
for classs in classesPath:
files = os.listdir(directory+os.sep+classs+os.sep)
for image in files:
if (not "db" in image and not ".directory" in image):
log.printMessage(("Reading:", directory+os.sep+classs+os.sep+image))
img = Image.open(directory+os.sep+classs+os.sep+image)
img = numpy.array(img)
if not inputType == DataUtil.INPUT_TYPE["Color"]:
img = grayImage(img,imageSize,False,"")
img = whiten(img)
img = resize(img,imageSize)
newFeatures = []
newFeatures.append(int(classNumber))
for x in img:
for y in x:
if inputType == DataUtil.INPUT_TYPE["Color"]:
colors = []
for c in y:
colors.append(c)
newFeatures.append(colors)
else:
newFeatures.append(y)
featuresSetGray.append(newFeatures)
classNumber = classNumber+1
print "Total features:", len(featuresSetGray)
DataUtil.writeSingleFile(featuresSetGray,featuresDirectory+os.sep+featuresName, inputType == DataUtil.INPUT_TYPE["Color"])