git clone https://github.com/wayneweiqiang/DeepDenoiser.git
cd DeepDenoiser
conda env update -f=env.yml -n base
conda env create -f env.yml
conda activate deepdenoiser
Located in directory: model/190614-104802
See details in the notebook: example_interactive.ipynb
See details in the notebook: example_batch_prediction.ipynb
Required: two csv files for signal and noise, corresponding directories of the npz files.
The csv file contains four columns: "fname", "itp", "channels"
The npz file contains four variable: "data", "itp", "channels"
The shape of "data" variables has a shape of 9001 x 3
The variables "itp" is the data points of first P arrival times.
Note: In the demo data, for simplicity we use the waveform before itp as noise samples, so the train_noise_list is same as train_signal_list here.
python deepdenoiser/train.py --mode=train --train_signal_dir=./Dataset/train --train_signal_list=./Dataset/train.csv --train_noise_dir=./Dataset/train --train_noise_list=./Dataset/train.csv --batch_size=20
Please let us know of any bugs found in the code. Suggestions and collaborations are welcomed