private void namesFileButton_Click(object sender, EventArgs e)
{
string filePath = GetFilePath(filterString: "YOLO Names File (*.names)|*names", defaultFile: "coco.names", startPath: GetDir(namesTextBox.Text));
if (filePath != null)
{
namesTextBox.Text = filePath;
namesTextBox.SelectionStart = namesTextBox.Text.Length;
namesTextBox.SelectionLength = 0;
try
{
igorCore.igorCore igor = new igorCore.igorCore();
igor.ThereWolfThereCastle(cfg: "", weights: "", namesTextBox.Text);
numberOfObjectsLabel.Text = "The selected model will code images for " + igor.theBags.Count() + " different object classes.";
igor.WhatHump();
gotNames = true;
}
catch
{
numberOfObjectsLabel.Text = "There was a problem with your \"names\" file.";
namesTextBox.Text = "";
MessageBox.Show("There was a problem with your \"names\" file. Igor was unable to read the list of objects contained within. " +
"Please check that your file is correctly formatted and is accessible for Igor to read.", "I Ain't Got No Body", MessageBoxButtons.OK, MessageBoxIcon.Error);
gotNames = false;
}
}
else
{
namesTextBox.Text = "";
numberOfObjectsLabel.Text = "There is no \"names\" file selected.";
gotNames = false;
}
}
public igorForm()
{
InitializeComponent();
igorCore.igorCore igor = new igorCore.igorCore();
gotFileIn = false;
gotFileOut = false;
RecursiveSearchCheckbox.Checked = true;
try
{
string currentDir = Directory.GetCurrentDirectory();
igor.Werewolf(Path.Combine(currentDir, "modelDat", "modelSelect.txt"));
cfgTextBox.Text = Path.GetFullPath(igor.modelDetailsCfg);
weightsTextBox.Text = Path.GetFullPath(igor.modelDetailsWeights);
namesTextBox.Text = Path.GetFullPath(igor.modelDetailsNames);
cfgTextBox.SelectionStart = cfgTextBox.Text.Length;
cfgTextBox.SelectionLength = 0;
weightsTextBox.SelectionStart = weightsTextBox.Text.Length;
weightsTextBox.SelectionLength = 0;
namesTextBox.SelectionStart = namesTextBox.Text.Length;
namesTextBox.SelectionLength = 0;
numberOfObjectsLabel.Text = "The selected model will code images for " + igor.theBags.Count() + " different object classes.";
igor.WalkThisWay();
if (igor.gpuAvailable)
{
gpuLabel.BackColor = Color.LightGreen;
cpuLabel.BackColor = Color.DarkGray;
}
else
{
cpuLabel.BackColor = Color.LightGreen;
gpuLabel.BackColor = Color.Red;
}
gotCfg = true;
gotWeight = true;
gotNames = true;
}
catch (Exception ex)
{
gotCfg = false;
gotWeight = false;
gotNames = false;
}
finally
{
igor.WhatHump();
}
}
private void backgroundWorker_DoWork(object sender, DoWorkEventArgs e)
{
Bags bags = (Bags)e.Argument;
igorCore.igorCore igor = new igorCore.igorCore();
try
{
igor.ThereWolfThereCastle(bags.GetCfg(), bags.GetWeight(), bags.GetNames());
igor.WalkThisWay();
}
catch (Exception ex)
{
MessageBox.Show("There was a problem initializing your YOLO model. Please share the following information with the software developer:" + Environment.NewLine + ex.ToString(),
"Damn Your Eyes!", MessageBoxButtons.OK, MessageBoxIcon.Error);
return;
}
var inputFiles = Directory.EnumerateFiles(bags.GetFolderIn(), searchPattern: "*.*", searchOption: bags.getDirDepth())
.Where(s => s.EndsWith(".jpg") || s.EndsWith(".png"));
using (FileStream fileStream = new FileStream(bags.GetFileOut(), FileMode.Append, FileAccess.Write, FileShare.Read))
using (StreamWriter streamWriter = new StreamWriter(fileStream, System.Text.Encoding.UTF8))
{
string headerRow = "\"Filename\"";
for (int i = 0; i < igor.bagCount; i++)
{
headerRow += ",\"" + igor.theBags[i] + "\"";
}
streamWriter.WriteLine(headerRow);
foreach (string file in inputFiles)
{
if (bgworker.CancellationPending)
{
break;
}
bgworker.ReportProgress(0, file.Replace(bags.GetFolderIn(), ""));
Dictionary <string, int> imageObjects = new Dictionary <string, int>();
for (int i = 0; i < igor.bagCount; i++)
{
imageObjects.Add(igor.theBags[i], 0);
}
try
{
List <string> items = igor.Blücher(file);
foreach (string item in items)
{
imageObjects[item]++;
}
#region build/write output
StringBuilder outputRow = new StringBuilder();
//add the filename to the output
outputRow.Append("\"" + file.Replace(bags.GetFolderIn(), "").Replace("\"", "\"\"") + "\"");
for (int i = 0; i < igor.bagCount; i++)
{
outputRow.Append(',');
outputRow.Append(imageObjects[igor.theBags[i]]);
}
streamWriter.WriteLine(outputRow.ToString());
#endregion
}
catch (Exception ex)
{
using (FileStream fs = new FileStream(bags.GetFileOut() + ".log", FileMode.Create, FileAccess.Write, FileShare.Read))
using (StreamWriter logWriter = new StreamWriter(fileStream, System.Text.Encoding.UTF8))
{
logWriter.WriteLine("================================================");
logWriter.WriteLine("Error processing " + file);
logWriter.WriteLine(Environment.NewLine);
logWriter.WriteLine(ex.ToString());
}
}
}
}
igor.WhatHump();
}
static void Main(string[] args)
{
string execpath = Directory.GetCurrentDirectory();
//Console.WriteLine(args.Length.ToString());
//foreach (string arg in args) Console.WriteLine(arg);
//Console.WriteLine(args.Length.ToString());
//Thread.Sleep(3000);
//print the header
PrintHeader();
//bring an Igor() into the mix. This wakes him up,
//tells him where the model details are, etc.
igorCore.igorCore igor = new igorCore.igorCore();
PrintStrongLine();
Console.ForegroundColor = exampleColor;
try
{
igor.Werewolf(Path.Combine(execpath, "modelDat", "modelSelect.txt"));
}
catch (Exception ex)
{
WriteText(ex.ToString(), errColor);
EndProg();
}
Console.ForegroundColor = neutralColor;
Thread.Sleep(1000);
PrintStrongLine();
int numObjectCategories = igor.theBags.Count();
bool ableToLog = false;
//double check that we have the right number of arguments
#region check arguments
if (args.Length != 2)
{
WriteText("Igor requires 2 arguments: the input folder and", errColor);
WriteText("output file location. For example:", errColor);
WriteText(" igor.exe \"C:\\imagefiles\" \"C:\\OutputData.csv\" ", exampleColor);
Console.WriteLine();
WriteText("In some cases, the error is caused by unintentionally ending your", errColor);
WriteText("input folder parameter with an escape. For example:", errColor);
WriteText(" igor.exe \"C:\\imagefiles\\\" \"C:\\OutputData.csv\" ", exampleColor);
Console.WriteLine();
WriteText("Note the extra \"\\\" that is is not in the original example.", errColor);
EndProg();
}
try
{
if (!Directory.Exists(args[0]))
{
WriteText("Your input folder does not appear to be valid.", errColor);
EndProg();
}
}
catch (Exception ex)
{
PrintStrongLine();
Console.WriteLine(ex.ToString(), errColor);
PrintStrongLine();
WriteText("Your input parameters appear to be incorrect. For example:", errColor);
WriteText(" igor.exe \"C:\\imagefiles\" \"C:\\OutputData.csv\" ", exampleColor);
EndProg();
}
#region try to initialize file
try
{
// Create the file, or overwrite if the file exists.
using (FileStream fileStream = new FileStream(args[1], FileMode.Create, FileAccess.Write, FileShare.Read))
using (StreamWriter streamWriter = new StreamWriter(fileStream, System.Text.Encoding.UTF8))
{
string headerRow = "\"Filename\"";
for (int i = 0; i < numObjectCategories; i++)
{
headerRow += ",\"" + igor.theBags[i] + "\"";
}
streamWriter.WriteLine(headerRow);
}
}
catch (Exception ex)
{
PrintStrongLine();
Console.WriteLine(ex.ToString(), errColor);
PrintStrongLine();
WriteText("Your output file does not appear to be valid. Please check to ensure that", errColor);
WriteText("the file location correct and that the file is not open in another program.", errColor);
EndProg();
}
#endregion
// now that we've vetted the arguments, we can rename them to actual
//variables for better readability
string inputDir = args[0];
string outputFile = args[1];
string logFile = outputFile + ".log";
#region check to see if we can create a log
try
{
// Create the file, or overwrite if the file exists.
using (FileStream fileStream = new FileStream(logFile, FileMode.Create, FileAccess.Write, FileShare.Read))
using (StreamWriter streamWriter = new StreamWriter(fileStream, System.Text.Encoding.UTF8))
{
}
ableToLog = true;
}
catch (Exception ex)
{
//do nothing since ableToLog is false by default
}
#endregion
#endregion
//this actually instantiates the model. very important stuff.
Console.ForegroundColor = exampleColor;
igor.WalkThisWay();
Thread.Sleep(1000);
Console.ForegroundColor = neutralColor;
var inputFiles = Directory.EnumerateFiles(inputDir, searchPattern: "*.*", searchOption: SearchOption.AllDirectories)
.Where(s => s.EndsWith(".jpg") || s.EndsWith(".png"));;
using (FileStream fileStream = new FileStream(outputFile, FileMode.Append, FileAccess.Write, FileShare.Read))
using (StreamWriter streamWriter = new StreamWriter(fileStream, System.Text.Encoding.UTF8))
{
WriteText(" -> ...model has been loaded.");
foreach (string file in inputFiles)
{
Dictionary <string, int> imageObjects = new Dictionary <string, int>();
for (int i = 0; i < numObjectCategories; i++)
{
imageObjects.Add(igor.theBags[i], 0);
}
PrintFile(file, inputDir);
try
{
List <string> items = igor.Blücher(file);
foreach (string item in items)
{
imageObjects[item]++;
}
#region build/write output
//write our output
string headerRow = "\"Filename\"";
for (int i = 0; i < numObjectCategories; i++)
{
headerRow += ",\"" + igor.theBags[i] + "\"";
}
StringBuilder outputRow = new StringBuilder();
//add the filename to the output
outputRow.Append("\"" + file.Replace(inputDir, "").Replace("\"", "\"\"") + "\"");
for (int i = 0; i < numObjectCategories; i++)
{
outputRow.Append(',');
outputRow.Append(imageObjects[igor.theBags[i]]);
}
streamWriter.WriteLine(outputRow.ToString());
#endregion
}
catch (Exception ex)
{
WriteText(" -> There was an error processing this file.", errColor);
WriteText(" -> Please see .log file for details.", errColor);
if (ableToLog)
{
using (FileStream fs = new FileStream(logFile, FileMode.Append, FileAccess.Write, FileShare.Read))
using (StreamWriter logWriter = new StreamWriter(fs, System.Text.Encoding.UTF8))
{
logWriter.WriteLine(strongLine);
logWriter.WriteLine("Error processing " + file);
logWriter.WriteLine(Environment.NewLine);
logWriter.WriteLine(ex.ToString());
}
}
}
}
}
igor.WhatHump();
EndProg();
}
