preparation_allSets.py 727 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
#!/usr/bin/env python3

'''
preparation_main.py

Script to run for data preparation for a folder with sets from multiple
experiments.


__AUTHOR__: Michael Rudolf
__DATE__: 25-Mar-2019

'''

# Import the necessary modules
import importlib
import h5py
import numpy as np
import os
import shutil
import logging
import matplotlib.pyplot as plt

Michael Rudolf's avatar
Michael Rudolf committed
24 25
import modules.preparation as prp

26 27 28 29 30 31 32 33 34 35 36
importlib.reload(preparation)

# List of file paths, depending on where I run the script
folder_paths = {
    'office2': '/home/mrudolf/Documents/py_allSets_ML/'
}

folder = folder_paths['office2']

set_file_list = [folder+f for f in os.listdir(folder) if f.endswith('h5')]
cfg_file_list = [folder+f for f in os.listdir(folder) if f.endswith('ini')]