Пример #1
0
 def combobox_browse_callback(self, hObject):
     currentVal = hObject.currentText()
     file_list = utils.select_files(currentVal, depth='root')
     self.layout2_widgets[0].clear()
     for f in file_list:
         item = QListWidgetItem(f)
         file_type = os.path.splitext(f)[1]
         if not file_type:
             item.setIcon(QIcon(pyIcons.fileIO.folder))
         elif file_type == '.py':
             item.setIcon(QIcon(pyIcons.fileIO.python))
         elif file_type == '.html':
             item.setIcon(QIcon(pyIcons.fileIO.html))
         elif file_type == '.ipynb':
             item.setIcon(QIcon(pyIcons.fileIO.ipynb))
         elif file_type == '.yaml':
             item.setIcon(QIcon(pyIcons.fileIO.yaml))
         elif file_type == '.json':
             item.setIcon(QIcon(pyIcons.fileIO.json))
         elif set([file_type
                   ]).issubset(set(['.png', '.jpg', '.svg', '.gif'])):
             item.setIcon(QIcon(pyIcons.fileIO.image))
         elif set([file_type]).issubset(set(['.xlsx', '.csv', '.xls'])):
             item.setIcon(QIcon(pyIcons.fileIO.excel))
         elif set([file_type]).issubset(set(['.txt', '.md'])):
             item.setIcon(QIcon(pyIcons.fileIO.txt))
         else:
             pass
         self.layout2_widgets[0].addItem(item)
     self.layout2_widgets[0].setIconSize(QSize(16, 16))
Пример #2
0
 def combobox_ext_callback(self, hObject):
     current_dir = self.layout1_widgets[1].currentText()
     file_ext = hObject.currentText()
     file_list = utils.select_files(current_dir,
                                    ext=[file_ext],
                                    depth='root')
     self.set_browser_listbox(self.layout2_widgets[0], file_list)
Пример #3
0
def get_last_model(**kwargs):
    log_dir = get_varargin(kwargs, 'log_dir', cfg.FILEIO.LOG_DIR)
    and_key = get_varargin(kwargs, 'and_key', ['.ckpt'])
    or_key = get_varargin(kwargs, 'or_key', None)
    # List .h5 files in model_dir directory
    file_list = utils.select_files(log_dir, and_key=and_key, or_key=or_key)
    return sorted(file_list)[-1]
Пример #4
0
 def on_epoch_end(self, epoch, logs=None):
     self.progress.current = epoch+1
     self.progress()
     logging.info('loss: {:.4f} - accuracy: {:.4f}'\
         ' - val_loss: {:.4f} - val_accuracy: {:.4f}'\
         .format(logs['loss'], logs['accuracy'],
                 logs['val_loss'], logs['val_accuracy']))
     if epoch+1 == self.params['epochs']:
         logger.logPC_usage()
         logger.log_nvidia_smi_info()
     # # Delete ckpts file, only keep n-latest files
     last_model = get_last_model()
     ckpts_dir = Path(last_model).parent
     ckpts_filelist = sorted(utils.select_files(ckpts_dir,and_key = ['.ckpt'])
                                     , reverse = True)
     epoch_list = sorted(np.unique([get_epoch_number(e) for e in ckpts_filelist]),
                         reverse = True)
     if len(epoch_list) > cfg.TRAIN.MAX_CKPTS_FILES:
         cut_epoch = epoch_list[cfg.TRAIN.MAX_CKPTS_FILES-1]
         for f in ckpts_filelist:
             if get_epoch_number(f) < cut_epoch:
                 os.remove(f)
Пример #5
0
 def combobox_browse_callback(self, hObject):
     currentVal = hObject.currentText()
     file_list = utils.select_files(currentVal, depth='root')
     self.set_browser_listbox(self.layout2_widgets[0], file_list)