def getDF(folder):
# get all the classes in the images folder
classes = ListDir(folder)
df_data = DataFrame()
for c in classes:
# Get a list of all images in the class folder.
imgs = ListDir(JoinPath(folder, c))
# Create the dataframe.
df2 = DataFrame(imgs, columns=['image_id'])
df2['class'] = c
df_data = df_data.append(df2)
return df_data
def GetSubDirs(Path_):
"""
Функция возвращает список поддиректорий.
@param Path_: Дирикторий.
@return: В случае ошибки возвращает None.
"""
try:
dir_list = [Path_ + '/' + path for path in ListDir(Path_)]
dir_list = [path for path in dir_list if IsDir(path)]
return dir_list
except:
log.fatal(u'Read subfolder list error: %s' % Path_)
return None
def GetFiles(Path_):
"""
Функция возвращает список файлов в директории.
@param Path_: Дирикторий.
@return: В случае ошибки возвращает None.
"""
try:
file_list = None
file_list = [Path_ + '/' + x.lower() for x in ListDir(Path_)]
file_list = [x for x in file_list if IsFile(x)]
return file_list
except:
log.fatal(u'Ошибка определения списка файлов директории %s' % Path_)
return None
def GetSubDirs(Path_):
"""
Функция возвращает список поддиректорий.
@param Path_: Дирикторий.
@return: В случае ошибки возвращает None.
"""
try:
dir_list = [Path_ + '/' + path for path in ListDir(Path_)]
dir_list = [path for path in dir_list if IsDir(path)]
return dir_list
except:
log.fatal(u'Ошибка определения списка поддиректорий директории <%s>' %
Path_)
return None
def IsSubDir(Dir1_, Dir2_):
"""
Функция проверяет, является ли директория Dir1_
поддиректорией Dir2_.
@return: Возвращает True/False.
"""
dir1 = AbsPath(Dir1_)
dir2 = AbsPath(Dir2_)
if dir1 == dir2:
return True
else:
sub_dirs = [
path for path in [dir2 + '\\' + name for name in ListDir(dir2)]
if IsDir(path)
]
for cur_sub_dir in sub_dirs:
find = IsSubDir(Dir1_, cur_sub_dir)
if find:
return find
return False
def GetFilesByExt(Path_, Ext_):
"""
Функция возвращает список всех файлов в директории с указанным расширением.
@param Path_: Путь.
@param Ext_: Расширение, например '.pro'.
@return: В случае ошибки возвращает None.
"""
file_list = None
try:
if Ext_[0] != '.':
Ext_ = '.' + Ext_
Ext_ = Ext_.lower()
file_list = [Path_ + '/' + file_name for file_name in ListDir(Path_)]
file_list = [
file_name for file_name in file_list
if IsFile(file_name) and (SplitExt(file_name)[1].lower() == Ext_)
]
return file_list
except:
log.fatal(
u'Ошибка определения списка файлов по расширению %s из директории %s: %s'
% (Ext_, Path_, file_list))
return None
TEST_FOLDER = args.eval_folder
TRAIN_FOLDER = args.train_folder
TMP_MODEL = "tmp_max.h5"
ROOT_DIR = os.getcwd()
IMG_NAME = MODEL_NAME + "_train.png"
KERAS_MODEL = MODEL_NAME + ".h5"
TFLITE_MODEL = MODEL_NAME + ".tflite"
LABEL_MODEL = MODEL_NAME + ".txt"
CONFUSION_NAME = MODEL_NAME + "_confusion.png"
# --- Global Vars --- #
# --- Setup the Dataset Generators --- #
## 1. create dataset for the training and testing images
## 2. Use the datasets to create generators to pipeline to TensorFlow
classes = ListDir(TEST_FOLDER)
classes.sort()
def getDF(folder):
# get all the classes in the images folder
classes = ListDir(folder)
df_data = DataFrame()
for c in classes:
# Get a list of all images in the class folder.
imgs = ListDir(JoinPath(folder, c))
# Create the dataframe.
df2 = DataFrame(imgs, columns=['image_id'])
df2['class'] = c