在tensorflow对象检测API中，有没有办法知道一个对象检测模型有多少个参数？

import tensorflow as tf

# 加载对象检测模型并创建会话
detection_graph = tf.Graph()
with detection_graph.as_default():
    od_graph_def = tf.GraphDef()
    with tf.gfile.GFile('path/to/model.pb', 'rb') as fid:
        serialized_graph = fid.read()
        od_graph_def.ParseFromString(serialized_graph)
        tf.import_graph_def(od_graph_def, name='')

    sess = tf.Session(graph=detection_graph)

# 获取所有可训练的变量列表
trainable_vars = tf.trainable_variables()

# 计算参数数量
total_params = 0
for var in trainable_vars:
    shape = var.get_shape()
    var_params = 1
    for dim in shape:
        var_params *= dim.value
    total_params += var_params

print("对象检测模型的参数数量为:", total_params)