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

updates

This commit is contained in:
Glenn Jocher 2018-12-15 21:06:39 +01:00
parent b079c1b10c
commit b52a49cf12
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models.py
View File

@ -232,6 +232,7 @@ class Darknet(nn.Module):
def __init__(self, cfg_path, img_size=416): def __init__(self, cfg_path, img_size=416):
super(Darknet, self).__init__() super(Darknet, self).__init__()
self.module_defs = parse_model_config(cfg_path) self.module_defs = parse_model_config(cfg_path)
self.module_defs[0]['height'] = img_size self.module_defs[0]['height'] = img_size
self.hyperparams, self.module_list = create_modules(self.module_defs) self.hyperparams, self.module_list = create_modules(self.module_defs)