diff --git a/Use_yolov4_to_train_your_own_data.md b/Use_yolov4_to_train_your_own_data.md
index c60204b8..f1aa1d0d 100644
--- a/Use_yolov4_to_train_your_own_data.md
+++ b/Use_yolov4_to_train_your_own_data.md
@@ -59,9 +59,9 @@ tensorboard --logdir log --host 192.168.212.75 --port 6008
 # 5. 验证
 
 ```
-python model.py 3 weight/Yolov4_epoch166_coins.pth data/coin2.jpg data/coins.names
+python models.py 3 weight/Yolov4_epoch166_coins.pth data/coin2.jpg 608 608 data/coins.names
 
-python model.py num_classes weightfile imagepath namefile
+python models.py num_classes weightfile imagepath height width namefile
 ```
 coins.names
 ```
diff --git a/models.py b/models.py
index cc4a275c..3cca101e 100644
--- a/models.py
+++ b/models.py
@@ -464,12 +464,12 @@ 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')
+        print('  python models.py num_classes weightfile imgfile height width namefile')
 
     model = Yolov4(yolov4conv137weight=None, n_classes=n_classes, inference=True)
 
@@ -487,6 +487,7 @@ def forward(self, input):
     # Optional inference sizes:
     #   Hight in {320, 416, 512, 608, ... 320 + 96 * n}
     #   Width in {320, 416, 512, 608, ... 320 + 96 * m}
+    print(width, height)
     sized = cv2.resize(img, (width, height))
     sized = cv2.cvtColor(sized, cv2.COLOR_BGR2RGB)