public static void ToTexture(DatasetFloat set, int imgIndex, Texture2D tex)
{
var colors = new Color[ImgDims];
for (int y = 0; y < Height; y++)
{
for (int x = 0; x < Width; x++)
{
float pix = set.Images[imgIndex * ImgDims + y * Height + x];
colors[y * Height + x] = new Color(pix, pix, pix, 1f);
}
}
tex.SetPixels(0, 0, Width, Height, colors);
tex.Apply(false);
}
private static DatasetFloat LoadAsFloats(string imgPath, string lblPath)
{
var lblReader = new BigEndianBinaryReader(new FileStream(Path.Combine(Folder, lblPath), FileMode.Open));
var imgReader = new BigEndianBinaryReader(new FileStream(Path.Combine(Folder, imgPath), FileMode.Open));
lblReader.ReadInt32();
lblReader.ReadInt32();
int magicNum = imgReader.ReadInt32();
int NumImgs = imgReader.ReadInt32();
int Rows = imgReader.ReadInt32();
int Cols = imgReader.ReadInt32();
int ImgDims = Rows * Cols;
Debug.Log("MNIST: Loading " + imgPath + ", Imgs: " + NumImgs + " Rows: " + Rows + " Cols: " + Cols);
var set = new DatasetFloat(NumImgs);
for (int i = 0; i < NumImgs; i++)
{
byte lbl = lblReader.ReadByte();
set.Labels[i] = (int)lbl;
}
for (int i = 0; i < NumImgs; i++)
{
// Read order flips images axes to Unity style
for (int y = 0; y < Cols; y++)
{
for (int x = 0; x < Rows; x++)
{
byte pix = imgReader.ReadByte();
set.Images[i * ImgDims + (Cols - 1 - y) * Rows + x] = pix / 256f;
}
}
}
lblReader.Close();
imgReader.Close();
return(set);
}
public static void GetBatch(NativeArray <int> batch, DatasetFloat set, ref Rng rng)
{
// Todo: can transform dataset to create additional variation
UnityEngine.Profiling.Profiler.BeginSample("GetBatch");
if (set.Indices.Count < batch.Length)
{
set.Indices.Clear();
for (int i = 0; i < set.NumImgs; i++)
{
set.Indices.Add(i);
}
Ramjet.Utils.Shuffle(set.Indices, ref rng);
}
for (int i = 0; i < batch.Length; i++)
{
batch[i] = set.Indices[set.Indices.Count - 1];
set.Indices.RemoveAt(set.Indices.Count - 1);
}
UnityEngine.Profiling.Profiler.EndSample();
}
public static void LoadFloatData()
{
TrainFloats = LoadAsFloats(TrainImagePath, TrainLabelPath);
TestFloats = LoadAsFloats(TestImagePath, TestLabelPath);
}