How to get model summary

[ThanhNX0401]

i try to run the code but it got error no matter what model i run:

import tensorflow as tf
from doctr.models import linknet_resnet50
model = linknet_resnet50(pretrained=True)
input_tensor = tf.random.uniform(shape=[1, 1024, 1024, 3], maxval=1, dtype=tf.float32)
out = model(input_tensor)
model.summary()

This is the error: AttributeError: 'Activation' object has no attribute 'get_output_at'

[felixdittrich92]

Hi @ThanhNX0401 👋,

The issue is that you run with Keras v3 but docTR depends on Keras v2

Ensure that the following is installed:

"tf-keras>=2.15.0,<3.0.0"

pip3 install tf-keras

For safety you can put the following on top of your script before all other imports:

from doctr.file_utils import ensure_keras_v2

ensure_keras_v2()

Then your script should run as expected.

Hope this helps.

Best,
Felix