Implementation notes - AsyDynamics/keras-yolo2 Wiki

Tiny yolo

MobileNet


Layer (type) Output Shape Param # Connected to

input_1 (InputLayer) (None, 416, 416, 3) 0


model_1 (Model) (None, 13, 13, 1024) 3228864 input_1[0][0]


DetectionLayer (Conv2D) (None, 13, 13, 30) 30750 model_1[1][0]


reshape_1 (Reshape) (None, 13, 13, 5, 6) 0 DetectionLayer[0][0]


input_2 (InputLayer) (None, 1, 1, 1, 10, 0


lambda_1 (Lambda) (None, 13, 13, 5, 6) 0 reshape_1[0][0]
input_2[0][0]

Total params: 3,259,614 Trainable params: 3,237,726 Non-trainable params: 21,888