fix deprecation: torch.meshgrid

Author: clibdev

README.md

@@ -1,6 +1,8 @@
 # Fork of [deepcam-cn/yolov5-face](https://github.com/deepcam-cn/yolov5-face)
 
 * Added [requirements.txt](requirements.txt) file.
+* Fixed deprecations:
+  * UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument.
 
 # Installation
 

models/yolo.py

@@ -111,7 +111,7 @@ def forward(self, x):
 
     @staticmethod
     def _make_grid(nx=20, ny=20):
-        yv, xv = torch.meshgrid([torch.arange(ny), torch.arange(nx)])
+        yv, xv = torch.meshgrid([torch.arange(ny), torch.arange(nx)], indexing='ij')
         return torch.stack((xv, yv), 2).view((1, 1, ny, nx, 2)).float()
 
     def _make_grid_new(self,nx=20, ny=20,i=0):