diff --git a/birdnet_analyzer/search/utils.py b/birdnet_analyzer/search/utils.py index 35fce39ad..a31748f1b 100644 --- a/birdnet_analyzer/search/utils.py +++ b/birdnet_analyzer/search/utils.py @@ -86,7 +86,7 @@ def get_query_embedding( else: sig_splits = audio.split_signal(sig, rate, sig_length, crop_overlap, sig_minlen) - return model_utils.get_embeddings_array(sig_splits) + return model_utils.get_embeddings_array(sig_splits, n_workers=1) def get_search_results( @@ -145,7 +145,7 @@ def get_search_results( if use_ann: sorted_results = _search_ann_ip(db, embedding, n_results) else: - results, scores = brutalism.threaded_brute_search( + results = brutalism.threaded_brute_search( db, embedding, n_results, diff --git a/birdnet_analyzer/train/utils.py b/birdnet_analyzer/train/utils.py index 39d8d3436..737414b02 100644 --- a/birdnet_analyzer/train/utils.py +++ b/birdnet_analyzer/train/utils.py @@ -749,7 +749,7 @@ def generate_splits( if metrics: import csv - eval_file_path = os.path.join(output, "evaluation.csv") + eval_file_path = output + "_evaluation.csv" with open(eval_file_path, "w", newline="") as f: writer = csv.writer(f)