Ivar Reukers ivarreukers
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| public class FaceNet { | |
| private static final Logger log = LoggerFactory.getLogger(FaceNet.class); | |
| public static int numClasses = 40; | |
| public static int batchSize = 15; | |
| public static int nEpochs = 20; | |
| public static int trainPerc = 80; | |
| public static String DATA_DIR = "/Users/ivaro18/Downloads/ATTFaceNet"; | |
| public static final String[] allowedFormats = BaseImageLoader.ALLOWED_FORMATS; | |
| public static final Random rng = new Random(13); |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| protected static final long seed = 12345; | |
| protected static final int numClasses = 41; | |
| protected static final int nEpochs = 20; | |
| private static TransferLearningHelper transferLearningHelper; | |
| private static ComputationGraph vgg16transfer, vgg16; | |
| public static void main(String args[]) { | |
| ZooModel zooModel = new VGG16(); | |
| vgg16 = (ComputationGraph) zooModel.initPretrained(PretrainedType.VGGFACE); | |
| log.info(vgg16.summary()); |