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

updates

This commit is contained in:
Glenn Jocher 2019-02-19 22:49:47 +01:00
parent 0772ebf7c9
commit 44333ddd9f
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -101,9 +101,7 @@ class Upsample(nn.Module):
class YOLOLayer(nn.Module): class YOLOLayer(nn.Module):
def __init__(self, anchors, nC, img_size, yolo_layer, cfg): def __init__(self, anchors, nC, img_size, yolo_layer, cfg):
# TODO: img_size from hyperparams in cfg file, NOT from parser. Make dynamic
super(YOLOLayer, self).__init__() super(YOLOLayer, self).__init__()
nA = len(anchors) nA = len(anchors)