Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2020-03-09 10:46:59 -07:00
parent 071d4113f6
commit cd76a1a982
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -189,7 +189,7 @@ class YOLOLayer(nn.Module):
self.ny = 0 # initialize number of y gridpoints self.ny = 0 # initialize number of y gridpoints
self.arc = arc self.arc = arc
if ONNX_EXPORT: # grids must be computed in __init__ if ONNX_EXPORT:
stride = [32, 16, 8][yolo_index] # stride of this layer stride = [32, 16, 8][yolo_index] # stride of this layer
nx = img_size[1] // stride # number x grid points nx = img_size[1] // stride # number x grid points
ny = img_size[0] // stride # number y grid points ny = img_size[0] // stride # number y grid points