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)); }