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