public TrainingSet(string name, TrainingSettings training, TrainingDataSettings trainingData, TrainingNetworkSettings network)
{
Name = name;
Training = training;
TrainingData = trainingData;
Network = network;
}
// MARK: Training Settings Methods
private void EditTrainingSettings(TrainingSettings dialog)
{
int saveIters = int.Parse(dialog.endItersTextBox.Text);
CurrentProject.SaveIters = (saveIters / 5).ToString(); //save every 20%
CurrentProject.EndIters = dialog.endItersTextBox.Text;
CurrentProject.GlobalScale = dialog.GlobalScaleSlider.Value;
UpdateVGLConfig();
UpdateFramesToExtract();
}
// MARK: Training Methods
private void TrainButton_Click(object sender, RoutedEventArgs e) //start training
{
BarInteraction();
if (CurrentProject.IsTrained)
{
MessageBox.Show("The network has already been trained. If you choose to train again the previous network will be overwritten.", "Overwrite?");
}
TrainingSettings settingsDialog = new TrainingSettings();
settingsDialog.endItersTextBox.Text = CurrentProject.EndIters;
settingsDialog.GlobalScaleSlider.Value = CurrentProject.GlobalScale;
settingsDialog.GlobalScaleNumberText.Text = CurrentProject.GlobalScale.ToString();
if (settingsDialog.ShowDialog() == true)
{
EditTrainingSettings(settingsDialog);
CreateTrainingDataset();
}
else
{
EnableInteraction();
}
}
