public void SłownikEtykiet_ShouldAutoLabelFiles()
{
var etykiety = SłownikEtykiet.AutoLabel(new[] {
@"0_A.txt",
@"1_B.txt",
@"0_A.txt",
@"1_C.txt",
@"2_C.txt",
});
etykiety.Etykiety.Count().ShouldBe(3);
var join = string.Join("", etykiety.Etykiety);
join.ShouldBe("ABC");
}
public void LabelWriter_ShouldExportLabelsToFile()
{
var files = Directory.GetFiles("Samples", "*.png");
files.Length.ShouldBe(2);
var labels = SłownikEtykiet.AutoLabel(files);
labels.Etykiety.Count().ShouldBe(2);
var writer = new LabelWriter(labels);
var records = writer.Zapisz("Labels.txt", files);
records.Count().ShouldBe(2);
var first = records.First();
first.ShouldStartWith("I: 1");
var last = records.Last();
last.ShouldStartWith("X: 1");
File.Exists("Labels.txt").ShouldBe(true);
}
void EtykietujPliki(string folder, string labelsPath = null)
{
var files = Directory.GetFiles(folder, "*.png", SearchOption.AllDirectories);
var labels = new SłownikEtykiet();
if (File.Exists(labelsPath))
{
Console.WriteLine("Wczytywanie słownika etykiet z pliku " + labelsPath);
labels = SłownikEtykiet.Wczytaj(labelsPath);
}
else
{
Console.WriteLine("Generowanie słownika etykiet z " + files.Length + " plik(i)...");
labels = SłownikEtykiet.AutoLabel(files);
}
Console.WriteLine("Etykiety: {0}", labels.Etykiety.Count());
foreach (var e in labels.Etykiety)
{
Console.WriteLine(e + ": " + string.Join(",", labels.LabelToBinaryArray(e)));
}
var fileName = "LabelFeatures_T" + DateTime.Now.ToShortDateString() + ".txt";
Console.WriteLine("Etykietowanie " + files.Length + " plików -> " + fileName);
var writer = new FeatureWriter(labels)
{
UseColor = Settings.Default.Color
};
writer.Zapisz(fileName, files);
var labelsWriter = new LabelWriter(labels);
var labelPath = "LabelFeatures_LabelsT" + DateTime.Now.ToShortDateString() + ".txt";
Console.WriteLine("Eksportowanie etykiet do pliku -> " + labelPath);
var records = labelsWriter.Zapisz(labelPath, files);
foreach (var record in records)
{
Console.WriteLine(record);
}
}