I wonder if there is away when you call model.summary() it display it in such away there is an index for for each layer. For example like this:
1 input_image (InputLayer)       [(None, 128, 128, 3  0           []                               
                                )]                                                                
                                                                                                  
2 conv1_pad (ZeroPadding2D)      (None, 134, 134, 3)  0           ['input_image[0][0]']            
                                                                                                  
3 conv1_conv (Conv2D)            (None, 64, 64, 64)   9472        ['conv1_pad[0][0]']