Skip to content

Commit 267a040

Browse files
authored
Merge pull request #305 from kepler62f/master
data type for sys.argv
2 parents bc05533 + 533cb3c commit 267a040

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -464,9 +464,9 @@ def forward(self, input):
464464
n_classes = int(sys.argv[1])
465465
weightfile = sys.argv[2]
466466
imgfile = sys.argv[3]
467-
height = sys.argv[4]
467+
height = int(sys.argv[4])
468468
width = int(sys.argv[5])
469-
namesfile = int(sys.argv[6])
469+
namesfile = sys.argv[6]
470470
else:
471471
print('Usage: ')
472472
print(' python models.py num_classes weightfile imgfile namefile')

0 commit comments

Comments
 (0)