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

add stride order reversal for c53*.cfg

This commit is contained in:
Glenn Jocher 2020-05-15 14:40:14 -07:00
parent 6fe67595cb
commit 20891926c9
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

@ -93,7 +93,7 @@ def create_modules(module_defs, img_size, cfg):
elif mdef['type'] == 'yolo':
yolo_index += 1
stride = [32, 16, 8] # P5, P4, P3 strides
if 'panet' in cfg or 'yolov4' or 'cd53' in cfg: # stride order reversed
if any(x in cfg for x in ['panet', 'yolov4', 'cd53']): # stride order reversed
stride = list(reversed(stride))
layers = mdef['from'] if 'from' in mdef else []
modules = YOLOLayer(anchors=mdef['anchors'][mdef['mask']], # anchor list