stash

samedifflambda.java

    public class ScaleSeventeenPercent extends SameDiffLambdaLayer {
        @Override
        public SDVariable defineLayer(SameDiff sd, SDVariable x) { return x.mul(0.17); }

        @Override
        public InputType getOutputType(int layerIndex, InputType inputType) { return inputType; }
    }

    public class ScaleTenPercent extends SameDiffLambdaLayer {
        @Override
        public SDVariable defineLayer(SameDiff sd, SDVariable x) { return x.mul(0.1); }

        @Override
        public InputType getOutputType(int layerIndex, InputType inputType) { return inputType; }
    }
    public class ScaleTwentyPercent extends SameDiffLambdaLayer {
        @Override
        public SDVariable defineLayer(SameDiff sd, SDVariable x) { return x.mul(0.2); }

        @Override
        public InputType getOutputType(int layerIndex, InputType inputType) { return inputType; }
    }
    public class ScaleHundredPercent extends SameDiffLambdaLayer {
        @Override
        public SDVariable defineLayer(SameDiff sd, SDVariable x) { return x.mul(1.0); }

        @Override
        public InputType getOutputType(int layerIndex, InputType inputType) { return inputType; }
    }

    public KerasFaceNetFeatureProvider() throws IOException, InvalidKerasConfigurationException, UnsupportedKerasConfigurationException {
        String simpleMlp = new ClassPathResource("keras/facenet/facenet_keras.h5").getFile().getPath();
        KerasLayer.registerLambdaLayer("Block35_1_ScaleSum", new ScaleSeventeenPercent());
        KerasLayer.registerLambdaLayer("Block35_2_ScaleSum", new ScaleSeventeenPercent());
        KerasLayer.registerLambdaLayer("Block35_3_ScaleSum", new ScaleSeventeenPercent());
        KerasLayer.registerLambdaLayer("Block35_4_ScaleSum", new ScaleSeventeenPercent());
        KerasLayer.registerLambdaLayer("Block35_5_ScaleSum", new ScaleSeventeenPercent());

        KerasLayer.registerLambdaLayer("Block17_1_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_2_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_3_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_4_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_5_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_6_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_7_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_8_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_9_ScaleSum", new ScaleTenPercent());
        KerasLayer.registerLambdaLayer("Block17_10_ScaleSum", new ScaleTenPercent());

        KerasLayer.registerLambdaLayer("Block8_1_ScaleSum", new ScaleTwentyPercent());
        KerasLayer.registerLambdaLayer("Block8_2_ScaleSum", new ScaleTwentyPercent());
        KerasLayer.registerLambdaLayer("Block8_3_ScaleSum", new ScaleTwentyPercent());
        KerasLayer.registerLambdaLayer("Block8_4_ScaleSum", new ScaleTwentyPercent());
        KerasLayer.registerLambdaLayer("Block8_5_ScaleSum", new ScaleTwentyPercent());
        KerasLayer.registerLambdaLayer("Block8_6_ScaleSum", new ScaleHundredPercent());

        model = KerasModelImport.importKerasModelAndWeights(simpleMlp);