avm/hassio-addons
1
0
Fork 0
mirror of https://github.com/alexbelgium/hassio-addons.git synced 2026-07-29 17:30:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Allow argument, or all files

Browse Source
This commit is contained in:
Alexandre
2024-05-27 11:11:40 +02:00
committed by GitHub
parent 167747019a
commit 52d0805121
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
birdnet-pi/rootfs/helpers/spectrum_analysis.py → birdnet-pi/rootfs/helpers/spectral_analysis.py
View File

@@ -3,19 +3,28 @@ import scipy.io.wavfile as wavfile
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import os import os
import glob import glob
import sys # Import the sys module
from utils.helpers import get_settings
# Dependencies /usr/bin/pip install numpy scipy matplotlib # Dependencies /usr/bin/pip install numpy scipy matplotlib
# Define the directory containing the WAV files # Define the directory containing the WAV files
input_directory = '/tmp/StreamData' conf = get_settings()
input_directory = os.path.join(conf['RECS_DIR'], 'StreamData')
output_directory = '/config' output_directory = '/config'
# Ensure the output directory exists # Ensure the output directory exists
if not os.path.exists(output_directory): if not os.path.exists(output_directory):
os.makedirs(output_directory) os.makedirs(output_directory)
# Get a list of all WAV files in the input directory # Check if a command-line argument is provided
wav_files = glob.glob(os.path.join(input_directory, '*.wav')) if len(sys.argv) > 1:
# If an argument is provided, use it as the file to analyze
wav_files = [sys.argv[1]]
else:
# If no argument is provided, analyze all WAV files in the directory
wav_files = glob.glob(os.path.join(input_directory, '*.wav'))
# Process each file # Process each file
for file_path in wav_files: for file_path in wav_files: