I have to evaluate my CNN model for Semantic Segmentation (when predicting) in terms of computational complexity, memory use and processing time. I work with Python (Keras).
Is there a way to count the number of operations performed in each layer or at least in the whole model?
I was taking a look to profiling libraries such as CProfile, but honestly, I am not sure of what I am getting.
Any help, please?
Thanks in advance!