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

updates

This commit is contained in:
Glenn Jocher 2019-08-17 14:15:27 +02:00
parent b8c870711f
commit b72fb74ad0
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

@ -96,7 +96,7 @@ class YOLOLayer(nn.Module):
def __init__(self, anchors, nc, img_size, yolo_index): def __init__(self, anchors, nc, img_size, yolo_index):
super(YOLOLayer, self).__init__() super(YOLOLayer, self).__init__()
self.anchors = torch.from_numpy(anchors) self.anchors = torch.Tensor(anchors)
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.nx = 0 # initialize number of x gridpoints self.nx = 0 # initialize number of x gridpoints