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 11:32:23 -07:00
parent b2fcfc573e
commit 6fe67595cb
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': elif mdef['type'] == 'yolo':
yolo_index += 1 yolo_index += 1
stride = [32, 16, 8] # P5, P4, P3 strides stride = [32, 16, 8] # P5, P4, P3 strides
if 'panet' in cfg or 'yolov4' in cfg: # stride order reversed if 'panet' in cfg or 'yolov4' or 'cd53' in cfg: # stride order reversed
stride = list(reversed(stride)) stride = list(reversed(stride))
layers = mdef['from'] if 'from' in mdef else [] layers = mdef['from'] if 'from' in mdef else []
modules = YOLOLayer(anchors=mdef['anchors'][mdef['mask']], # anchor list modules = YOLOLayer(anchors=mdef['anchors'][mdef['mask']], # anchor list