public ModelBuilderTest(string strName, int nDeviceID, EngineParameter.Engine engine)
: base(strName, null, nDeviceID)
{
m_engine = engine;
m_strName = strName;
m_strBaseDir = TestBase.GetTestPath("\\MyCaffe\\test_data", true, true);
}
public void TestLoadingFiles()
{
string strPath = TestBase.GetTestPath("\\MyCaffe\\test_data\\models", true, true);
string[] rgstrDir = Directory.GetDirectories(strPath);
foreach (string strDir in rgstrDir)
{
string[] rgstrFiles = Directory.GetFiles(strDir);
foreach (string strFile in rgstrFiles)
{
FileInfo fi = new FileInfo(strFile);
if (fi.Extension == ".prototxt")
{
Trace.WriteLine("Loading '" + fi.FullName + "...");
RawProto proto1 = RawProtoFile.LoadFromFile(fi.FullName);
string strProto1 = proto1.ToString();
RawProto proto2 = RawProto.Parse(strProto1);
string strProto2 = proto2.ToString();
Assert.AreEqual(strProto1, strProto2);
}
}
}
}
protected string getTestPath(string strItem, bool bPathOnly = false, bool bCreateIfMissing = false, bool bUserData = false)
{
return(TestBase.GetTestPath(strItem, bPathOnly, bCreateIfMissing, bUserData));
}
