用segnet网络训练后进行预测时出现AttributeError

AttributeError:module 'tensorflow. compat.V2.internal'has no atribute 'tf2'
求大神看看吧，难死孩子了

参考https://stackoverflow.com/questions/67696519/module-tensorflow-compat-v2-internal-has-no-attribute-tf2。
在导入时用了import keras?试试改成 tensorflow import keras,或者用
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
os.environ["SM_FRAMEWORK"] = "tf.keras"
from tensorflow import keras
import segmentation_models as sm