operationPath = "D:\pythonworkspace\TensorflowTraining\exercises\Shen\Practice\ACFdata\Src" os.chdir(operationPath) IR_database = os.path.join(os.path.dirname(operationPath), 'IR_database') IR_logs = os.path.join(os.path.dirname(operationPath), 'IR_logs') IR_model = os.path.join(os.path.dirname(operationPath), 'IR_Models') trainDir = os.path.join(IR_database, 'TrainingData', 'TrainFilted10\\') validateDir = os.path.join(IR_database, 'ValidateData', 'ValidateFilted6\\') testDir = os.path.join(IR_database, 'TestData', 'TestFilted6\\') trainLogs = os.path.join(IR_logs, 'IR_Trainlogs', TrainingPlan, DATE) testLogs = os.path.join(IR_logs, 'IR_testlogs', TrainingPlan, DATE) modelDir = os.path.join(IR_model, TrainingPlan, DATE) makeDir.mkdir(modelDir) #loadModelDir = os.path.join(IR_model,TrainingPlan,'2017-12-27-11-42','IR.ckpt-4999') #loadModelDir = os.path.join(IR_model,TrainingPlan,'2017-12-27-13-02','IR.ckpt-4999') #Input Parameter# IMAGE_WIDTH = 128 IMAGE_HEIGHT = 128 IMAGE_CHANNEL = 3 #Inference Parameter# size_in = 0 size_out = 0 CLASS_NUM = 2 #Training Parameter#
def image_augment(filename,angle=None,luminance=None,Contrast=None):
#img = Image.open(filename)
img = cv2.imread(filename)
name = filename.split('_')
path = os.getcwd()
currentdir = path.split('\\')
paraent_path = os.path.dirname(path)
if angle is not None:
ropath = os.path.join(paraent_path,currentdir[-1]+'_rotate')
rodirsuccess = mkdir(ropath)
for a in angle:
# img = img.rotate(a)
rows,cols,channels = img.shape
M = cv2.getRotationMatrix2D((cols/2,rows/2),a,1)
dst = cv2.warpAffine(img,M,(cols,rows))
rotation_name = name[0]+'_ro'+str(a)
for n in name[1:]:
rotation_name+=('_'+n)
save_path = os.path.join(ropath,rotation_name)
try:
# img.save(save_path)
cv2.imwrite(save_path,dst)
except IOError:
print("cannot rotation")
else:
#print("Don't do any rotation")
pass
if luminance is not None:
lupath = os.path.join(paraent_path,currentdir[-1]+'_luminance')
ludirsuccess = mkdir(lupath)
for l in luminance:
img = ImageEnhance.Brightness(img).enhance(l)
luminance_name = name[0]+'_lu'+str(l)
for n in name[1:]:
luminance_name+=('_'+n)
save_path = os.path.join(lupath,luminance_name)
try:
img.save(save_path)
except IOError:
print("cannot enhance luminance")
else:
#print("Don't do any lumiance enhancement")
pass
if Contrast is not None:
copath = os.path.join(paraent_path,currentdir[-1]+'_contrast')
codirsuccess = mkdir(copath)
for C in Contrast:
img = ImageEnhance.Contrast(img).enhance(C)
contrast_name = name[0]+'_co'+str(C)
for n in name[1:]:
contrast_name+=('_'+n)
save_path = os.path.join(copath,contrast_name)
try:
img.save(save_path)
except IOError:
print("cannot enhance contrast")
else:
#print("Don't do any contrast enhancement")
pass