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

multi_gpu multi_scale

This commit is contained in:
Glenn Jocher 2019-03-19 15:44:36 +02:00
parent 735b1a370b
commit bc989a0147
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
models.py
View File

@ -105,9 +105,8 @@ class YOLOLayer(nn.Module):
self.nA = len(anchors) # number of anchors (3) self.nA = len(anchors) # number of anchors (3)
self.nC = nC # number of classes (80) self.nC = nC # number of classes (80)
self.img_size = 0 self.img_size = 0
# self.nG, self.stride, self.grid_xy, self.anchor_vec, self.anchor_wh = \ self.nG, self.stride, self.grid_xy, self.anchor_vec, self.anchor_wh = \
# [], [], [], [], [] [], [], [], [], []
create_grids(self, 32, 1)
if ONNX_EXPORT: # grids must be computed in __init__ if ONNX_EXPORT: # grids must be computed in __init__
stride = [32, 16, 8][yolo_layer] # stride of this layer stride = [32, 16, 8][yolo_layer] # stride of this layer