if int(mode_input) == 1:
clean_run = input(
"Destroy existing resources and train from scratch " +
"(WARNING: this is memory intensive and may take considerable time)? Enter Y/N: "
)
while clean_run.upper() not in ['Y', 'N']:
clean_run = input(
"Please enter Y (clean run) or N (retrain existing resources): ")
if clean_run.upper() == 'Y':
file_manager = FileSystemManager(image_directory, model_directory)
file_manager.clean_run()
extract_dir = file_manager.extract_archive(source_archive)
file_manager.remove_files_except('.png')
file_manager.data_science_fs(category0='benign', category1='malignant')
file_manager.organise_files(extract_dir,
category_rules={
'benign': 'SOB_B_.*.png',
'malignant': 'SOB_M_.*.png'
})
elif clean_run.upper() == 'N':
pass
else:
sys_utils.graceful_exit()
