Using tf 2.0 api to calculate total flops and parameters

MACS.py

import tensorflow as tf
session = tf.compat.v1.Session()
graph = tf.compat.v1.get_default_graph()


with graph.as_default():
    with session.as_default():
        base_model = tf.keras.applications.MobileNetV2(weights='imagenet', input_tensor=tf.compat.v1.placeholder('float32', shape=(1,224,224,3)))
        run_meta = tf.compat.v1.RunMetadata()
        opts = tf.compat.v1.profiler.ProfileOptionBuilder.float_operation()

        # We use the Keras session graph in the call to the profiler.
        flops = tf.compat.v1.profiler.profile(graph=graph,
                                              run_meta=run_meta, cmd='op', options=opts)
        
        opts = tf.compat.v1.profiler.ProfileOptionBuilder.trainable_variables_parameter()
        params = tf.compat.v1.profiler.profile(graph, run_meta=run_meta, cmd='op', options=opts)