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);
}
