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

updates

This commit is contained in:
Glenn Jocher 2019-12-29 14:54:08 -08:00
parent 894218390b
commit d13312b751
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

@ -13,7 +13,7 @@ def create_modules(module_defs, img_size, arc):
hyperparams = module_defs.pop(0) hyperparams = module_defs.pop(0)
output_filters = [int(hyperparams['channels'])] output_filters = [int(hyperparams['channels'])]
module_list = nn.ModuleList() module_list = nn.ModuleList()
routs = [] # list of layers which rout to deeper layes routs = [] # list of layers which rout to deeper layers
yolo_index = -1 yolo_index = -1
for i, mdef in enumerate(module_defs): for i, mdef in enumerate(module_defs):