Defult option for CPOD and porpoise to click train detector.

Also a minor
This commit is contained in:
Jamie Mac 2022-02-22 10:36:55 +00:00
parent d5c79d9559
commit 4e59317882
4 changed files with 70 additions and 23 deletions

View File

@ -1,10 +1,12 @@
package clickTrainDetector.clickTrainAlgorithms.mht;
import PamguardMVC.PamDataBlock;
import clickTrainDetector.clickTrainAlgorithms.mht.mhtvar.BearingChi2VarParams;
import clickTrainDetector.clickTrainAlgorithms.mht.mhtvar.BearingChi2VarParams.BearingJumpDrctn;
import clickTrainDetector.clickTrainAlgorithms.mht.mhtvar.CorrelationChi2Params;
import clickTrainDetector.clickTrainAlgorithms.mht.mhtvar.IDIChi2Params;
import clickTrainDetector.clickTrainAlgorithms.mht.mhtvar.SimpleChi2VarParams;
import cpod.CPODClickDataBlock;
/**
* Default parameters for different species for the MHT click train detector.
@ -74,10 +76,11 @@ public class DefaultMHTParams {
/**
* Get default MHT parameters for a particular default species.
* @param pamDataBlock
*
* @return the default MHT parameters for giver DefaultMHTSpecies.
*/
public static MHTParams getDefaultMHTParams(DefaultMHTSpecies defaultSpecies) {
public static MHTParams getDefaultMHTParams(DefaultMHTSpecies defaultSpecies, PamDataBlock<?> pamDataBlock ) {
MHTParams mhtParams = new MHTParams();
@ -147,6 +150,46 @@ public class DefaultMHTParams {
break;
case PORPOISE:
if (pamDataBlock!=null && pamDataBlock.getClass().isAssignableFrom(CPODClickDataBlock.class)) {
chi2Settings[0]= new IDIChi2Params("ICI", "s", 0.2, 0.0002, SimpleChi2VarParams.SCALE_FACTOR_ICI);
((IDIChi2Params) chi2Settings[0]).minIDI = 0.001;
chi2Settings[1]= new SimpleChi2VarParams("Amplitude", "dB", 100, 0.6, SimpleChi2VarParams.SCALE_FACTOR_AMPLITUDE);
chi2Settings[2]= new BearingChi2VarParams("Bearing Delta", "\u00b0", Math.toRadians(0.2), Math.toRadians(0.01),
SimpleChi2VarParams.SCALE_FACTOR_BEARING); //RADIANS
chi2Settings[3]= new CorrelationChi2Params("Correlation");
simpleChi2Params.chi2Settings = chi2Settings;
simpleChi2Params.enable = new boolean[] {true, true, false, false, false, false};
simpleChi2Params.maxICI = 0.25;
simpleChi2Params.coastPenalty = 10;
simpleChi2Params.newTrackPenalty = 50;
simpleChi2Params.longTrackExponent = 0.01;
simpleChi2Params.lowICIExponent = 2;
simpleChi2Params.newTrackN = 3;
//the MHT kernel params
mhtKernelParams.maxCoast = 2;
mhtKernelParams.nHold = 100;
mhtKernelParams.nPruneBackStart = 15;
mhtKernelParams.nPruneback = 8;
mhtKernelParams.nPruneback = 8;
//set the two main parameter classes.
mhtParams.chi2Params = simpleChi2Params;
mhtParams.mhtKernal = mhtKernelParams;
}
else {
chi2Settings[0]= new IDIChi2Params("ICI", "s", 0.2, 0.0002, SimpleChi2VarParams.SCALE_FACTOR_ICI);
chi2Settings[1]= new SimpleChi2VarParams("Amplitude", "dB", 30, 1, SimpleChi2VarParams.SCALE_FACTOR_AMPLITUDE);
chi2Settings[2]= new BearingChi2VarParams("Bearing Delta", "\u00b0", Math.toRadians(0.2), Math.toRadians(0.01),
@ -169,6 +212,8 @@ public class DefaultMHTParams {
mhtParams.chi2Params = simpleChi2Params;
mhtParams.mhtKernal = mhtKernelParams;
}
break;
case SPERM_WHALE:

View File

@ -222,7 +222,7 @@ public class MHTSettingsPane extends SettingsPane<MHTParams> {
for (DefaultMHTSpecies speciesTypes: DefaultMHTSpecies.values()) {
menuItem = new MenuItem(DefaultMHTParams.getDefaultSpeciesName(speciesTypes));
menuItem.setOnAction((action)->{
setParams(DefaultMHTParams.getDefaultMHTParams(speciesTypes));
setParams(DefaultMHTParams.getDefaultMHTParams(speciesTypes, mhtClickTrainAlgorithm.getClickTrainControl().getParentDataBlock()));
});
speciesChoiceBox.getItems().add(menuItem);
}

View File

@ -92,6 +92,12 @@ public class GenericDLClassifier implements DLClassiferModel, PamSettings {
//System.out.println("PrepModel! !!!");
genericModelWorker.prepModel(genericModelParams, dlControl);
//set cusotm transforms in the model.
if (genericModelParams.dlTransfromParams!=null) {
//important to remkae transforms from params
genericModelParams.dlTransfroms = DLTransformsFactory.makeDLTransforms((ArrayList<DLTransfromParams>)genericModelParams.dlTransfromParams);
}
genericModelWorker.setModelTransforms(genericModelParams.dlTransfroms);

View File

@ -120,11 +120,7 @@ public class KetosWorker extends DLModelWorker<KetosResult> {
// System.out.println("Class name " + i + " " + dlParams.classNames[i]);
// }
// }
// ketosDLParams.classNames = dlControl.getClassNameManager().makeClassNames(ketosParams.classNames);
// if (ketosParams.classNames!=null) {
// for (int i = 0; i<ketosDLParams.classNames.length; i++) {
// System.out.println("Class name " + i + " " + ketosDLParams.classNames[i].className + " ID " + ketosDLParams.classNames[i].ID );