batdetect2/scripts
2023-04-07 16:26:01 -06:00
..
gen_dataset_summary_image.py Changed bat_detect to batdetect2 2023-04-07 11:24:22 -06:00
gen_spec_image.py fixed missing get_default_db_config function 2023-04-07 16:26:01 -06:00
gen_spec_video.py fixed missing get_default_db_config function 2023-04-07 16:26:01 -06:00
README.md added viz scripts 2022-12-14 18:19:52 +00:00
viz_helpers.py Changed bat_detect to batdetect2 2023-04-07 11:24:22 -06:00

This directory contains some scripts for visualizing the raw data and model outputs.

gen_spec_image.py: saves the model predictions on a spectrogram of the input audio file.
e.g.
python gen_spec_image.py ../example_data/audio/20170701_213954-MYOMYS-LR_0_0.5.wav ../models/Net2DFast_UK_same.pth.tar

gen_spec_video.py: generates a video showing the model predictions for a file. e.g.
python gen_spec_video.py ../example_data/audio/20170701_213954-MYOMYS-LR_0_0.5.wav ../models/Net2DFast_UK_same.pth.tar

gen_dataset_summary_image.py: generates an image displaying the mean spectrogram for each class in a specified dataset.
e.g.
python gen_dataset_summary_image.py --ann_file PATH_TO_ANN/australia_TRAIN.json PATH_TO_AUDIO/audio/ ../plots/australia/