test.py 212 字节
Newer Older
J
JiaQi Xu 已提交
1
from nets.yolo4 import yolo_body
J
JiaQi Xu 已提交
2 3 4 5 6 7
from keras.layers import Input
inputs = Input([416,416,3])
model = yolo_body(inputs,3,80)
model.summary()

for i,layer in enumerate(model.layers):
J
JiaQi Xu 已提交
8
    print(i,layer.name)