Merge pull request #305 from kepler62f/master

data type for sys.argv
Tianxiaomo · Dec 12, 2021 · 267a040 · 267a040
2 parents bc05533 + 533cb3c
commit 267a040
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models.py b/models.py
@@ -464,9 +464,9 @@ def forward(self, input):
         n_classes = int(sys.argv[1])
         weightfile = sys.argv[2]
         imgfile = sys.argv[3]
-        height = sys.argv[4]
+        height = int(sys.argv[4])
         width = int(sys.argv[5])
-        namesfile = int(sys.argv[6])
+        namesfile = sys.argv[6]
     else:
         print('Usage: ')
         print('  python models.py num_classes weightfile imgfile namefile')