axiom-data-science / OrcaCNN

Detect Orca in passive acoustic datasets
MIT License
18 stars 8 forks source link

Average predictions from the 4 ML models to find start and end times of orca calls #42

Open ZER-0-NE opened 3 years ago

ZER-0-NE commented 3 years ago

Currently, there are 4 models uploaded to help in the prediction of orca calls. Some of these are checkpoint models which were saved while training during the GSoC period.

The 4 models offer different accuracy or number of predicted 1-second calls for the same unsampled acoustic data. Unfortunately, at this point, there is no one model for all.

The idea here is to average the predictions from the base models so that we can be more accurate about the predicted orca calls and their duration of occurrence in the acoustic/input data.

Since the autonomous recordings or the input data is divided into 1-second chunks, this makes the whole process easier. For eg., if 3 of the 4 base models predict that there is an orca call at the 37th second of the input data, we consider that there is a high probability of an orca call at that time. To generalize this observation, we consider with high confidence the presence of an orca call only if two or more of the base models predict it.

Tossing all the predictions this way to a .csv file seems like a good idea at this point in time. In addition, there can be 2 sections in the csv file

Benjamintdk commented 2 years ago

Hi @ZER-0-NE,

I am interested in contributing to OrcaSound efforts and have cloned this repository to play around with it a little. May I know if this patch is still required? I can do one up quickly and submit a PR for your consideration.

Benjamintdk commented 2 years ago

Have created a PR #44 to resolve this issue.