Code cleanup and readme fixes

2021-04-09 13:17:41 +05:30
parent 37b2a5bd98
commit 1de804f98d
8 changed files with 37 additions and 26 deletions
@@ -25,7 +25,12 @@ int main(int argc, char *argv[]) {
    std::string net = "yolo4tiny_fp32.rt";
    if(argc > 1)
        net = argv[1]; 
-    std::string input = "..\..\..\demo\yolo_test.mp4";
+    #ifdef __linux__ 
+        std::string input = "../demo/yolo_test.mp4";
+    #elif _WIN32
+        std::string input = "..\\..\\..\\demo\\yolo_test.mp4";
+    #endif
+
    if(argc > 2)
        input = argv[2]; 
    char ntype = 'y';