public ImageFolderDatasetCreator( ClassificationConfig config, int imageWidth, int imageHeight) { _config = config; _imageWidth = imageWidth; _imageHeight = imageHeight; }
public static void Init( ClassificationConfig config, string configFile, string baseModelFile, string newModelFile, string trainImageFolder, string trainDatasetFile, string testImageFolder, string testDatasetFile) { CreateAndSaveModel(config, baseModelFile, newModelFile); CreateAndSaveDatasets(config, trainImageFolder, trainDatasetFile, testImageFolder, testDatasetFile); config.Save(configFile); }
public static void CreateAndSaveDatasets( ClassificationConfig config, string trainImageFolder, string trainDatasetFile, string testImageFolder, string testDatasetFile) { var datasetCreator = new ImageFolderDatasetCreator( config, IMAGE_WIDTH, IMAGE_HEIGHT); var dataFileCreator = new DataFileCreator(); var trainDataset = datasetCreator.GetDataset(trainImageFolder); dataFileCreator.CreateDataFile(trainDataset, trainDatasetFile); var testDataset = datasetCreator.GetDataset(testImageFolder); dataFileCreator.CreateDataFile(testDataset, testDatasetFile); }
public static void CreateAndSaveModel( ClassificationConfig config, string baseModelFile, string newModelFile) { const string featureNodeName = "features"; const string lastHiddenNodeName = "z.x"; const string predictionNodeName = "prediction"; var inputShape = new int[] { IMAGE_WIDTH, IMAGE_HEIGHT, IMAGE_DEPTH }; var device = DeviceDescriptor.CPUDevice; var model = CntkHelper.BuildTransferLearningModel( Function.Load(baseModelFile, device), featureNodeName, predictionNodeName, lastHiddenNodeName, inputShape, config.ClassCount, device); model.Save(newModelFile); }
private static void Init() { var config = new ClassificationConfig( new string[] { "Tiger", "Leopard", "Puma", "Lynx", "Lion" }); CatsClassificationInitializer.Init( config, FinalizePath(CONFIG_FILE), FinalizePath(BASE_MODEL_FILE), FinalizePath(NEW_MODEL_FILE), FinalizePath(TRAIN_IMAGE_FOLDER), FinalizePath(TRAIN_DATASET_FILE), FinalizePath(TEST_IMAGE_FOLDER), FinalizePath(TEST_DATASET_FILE)); }
public Form1() { InitializeComponent(); _modelWrapper = new CntkModelWrapper(TRAINED_MODEL_FILE, _device); _config = ClassificationConfig.Load(CONFIG_FILE); }