Skip to content

Commit 19f4268

Browse files
committed
Fixes missing connection when include_top=False
1 parent 5ac46ce commit 19f4268

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

‎densenet.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -627,6 +627,7 @@ def __create_fcn_dense_net(nb_classes, img_input, include_top, nb_dense_block=5,
627627
channel, row, col = input_shape
628628
else:
629629
row, col, channel = input_shape
630+
x = x_up
630631

631632
x = Reshape((row * col, nb_classes))(x)
632633
x = Activation(activation)(x)

0 commit comments

Comments
 (0)