frederik
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
import argparse
|
|
import datetime
|
|
import os
|
|
|
|
from utils.helpers import get_settings, MODEL_PATH
|
|
from utils.models import MDataModel1, MDataModel2
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description='Get list of species for a given location with BirdNET. Sorted by occurrence frequency.'
|
|
)
|
|
parser.add_argument('--threshold', type=float, default=0.05,
|
|
help='Occurrence frequency threshold. Defaults to 0.05.')
|
|
args = parser.parse_args()
|
|
|
|
conf = get_settings()
|
|
lat = conf.getfloat('LATITUDE')
|
|
lon = conf.getfloat('LONGITUDE')
|
|
week = datetime.datetime.today().isocalendar()[1]
|
|
|
|
print(f'Getting species list for {lat}/{lon}, Week {week}...', flush=True)
|
|
labels_path = os.path.join(MODEL_PATH, 'labels.txt')
|
|
with open(labels_path, 'r') as lfile:
|
|
labels = [line.strip() for line in lfile]
|
|
|
|
model = MDataModel1(args.threshold) if conf.getint('DATA_MODEL_VERSION') == 1 else MDataModel2(args.threshold)
|
|
model.set_meta_data(lat, lon, week)
|
|
species_list = model.get_species_list_details(labels)
|
|
|
|
for species in species_list:
|
|
print(f'{species[1]} - {species[0]:.4f}')
|
|
|
|
print("""
|
|
The above species list describes all the species that the model will attempt to detect.
|
|
If you don't see a species you want detected on this list, decrease your threshold.
|
|
|
|
NOTE: no actual changes to your BirdNET-Pi species list were made by running this command.
|
|
To set your desired frequency threshold, do it through the BirdNET-Pi web interface (Tools -> Settings -> Model)
|
|
""")
|