forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPbtxtParser.cs
More file actions
67 lines (56 loc) · 1.88 KB
/
PbtxtParser.cs
File metadata and controls
67 lines (56 loc) · 1.88 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
using Newtonsoft.Json;
using System;
using System.Collections.Generic;
using System.Text;
namespace TensorFlowNET.Examples.Utility
{
public class PbtxtItem
{
public string name { get; set; }
public int id { get; set; }
public string display_name { get; set; }
}
public class PbtxtItems
{
public List<PbtxtItem> items { get; set; }
}
public class PbtxtParser
{
public static PbtxtItems ParsePbtxtFile(string filePath)
{
string line;
string newText = "{\"items\":[";
using (System.IO.StreamReader reader = new System.IO.StreamReader(filePath))
{
while ((line = reader.ReadLine()) != null)
{
string newline = string.Empty;
if (line.Contains("{"))
{
newline = line.Replace("item", "").Trim();
//newText += line.Insert(line.IndexOf("=") + 1, "\"") + "\",";
newText += newline;
}
else if (line.Contains("}"))
{
newText = newText.Remove(newText.Length - 1);
newText += line;
newText += ",";
}
else
{
newline = line.Replace(":", "\":").Trim();
newline = "\"" + newline;// newline.Insert(0, "\"");
newline += ",";
newText += newline;
}
}
newText = newText.Remove(newText.Length - 1);
newText += "]}";
reader.Close();
}
PbtxtItems items = JsonConvert.DeserializeObject<PbtxtItems>(newText);
return items;
}
}
}