fixed topology file path issue

shoaib42 · Aug 9, 2021 · f27cb53 · f27cb53
1 parent 7a1f961
commit f27cb53
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/c/poseNet.cpp b/c/poseNet.cpp
@@ -321,11 +321,20 @@ poseNet* poseNet::Create( const commandLine& cmdLine )
 // loadTopology
 bool poseNet::loadTopology( const char* json_path, Topology* topology )
 {
+	const std::string path = locateFile(json_path);
+
+	if( path.length() == 0 )
+	{
+		LogError(LOG_TRT "poseNet -- failed to find topology file %s\n", json_path);
+		return false;
+	}
+
+	// load the json
 	nlohmann::json topology_json;
 
 	try
 	{
-		std::ifstream topology_file(json_path);
+		std::ifstream topology_file(path.c_str());
 		topology_file >> topology_json;
 	}
 	catch (...)