Closed
Description
Thanks for your nice work, there might exist some tiny flaw in model.py files when I exam the model summry:
input = Input(shape=(416,416,3))
model = yolo_body(input, 9, 80)
model.summary()
it will have key error about dict
so I changed it :
kk = None
try:
kk = kwargs['strides']
except:
kk = None
#some mistake here
darknet_conv_kwargs['padding'] = 'VALID' if kk==(2,2) else 'SAME'
and it works
I hope it will helpful
Metadata
Assignees
Labels
No labels