Python | Example | Directory 만들기 | Directory 내 file 존재 유무 확인

Directory 만들기 | Directory 내 file 존재 유무 확인

# 디렉토리 없을 경우 생성

import os

# --------------------------------------------------------------------------------
DIR_WEIGHTS = './kereas_conv_wts/'

if not os.path.exists(DIR_WEIGHTS):
    os.mkdir(DIR_WEIGHTS)
# --------------------------------------------------------------------------------

# 파일이 제대로 저장 됐는지 확인

# save wts
# --------------------------------------------------------------------------------
filename_wts = DIR_WEIGHTS + "keras_conv_seq_model_{0}.wts".format(datetime.now().strftime("%Y.%m.%d.%H:%M:%S"))
# --------------------------------------------------------------------------------
model.save_weights(filename_wts, overwrite=False)

# check save wts
wts_list = os.listdir(DIR_WEIGHTS)
if filename_wts.split('/')[-1] in wts_list:
    print '{0} is saved.'.format(filename_wts)
else:
    print '{0} is not saved. check your cource code'.format(filename_wts)

# temp