我在 python 中导入库时遇到问题

Question

代码：

import warnings
warnings.filterwarnings("ignore")
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import os
from PIL import Image
import matplotlib
from sklearn.model_selection import train_test_split
import tensorflow as tf
import keras
from keras.models import *
from keras.layers import *
from keras.callbacks import *
from tensorflow.keras.optimizers import *
from tqdm import tqdm
import cv2
import segmentation_models as sm
%load_ext tensorboard
import datetime
from keras.utils.vis_utils import plot_model


error:


AttributeError                            Traceback (most recent call last)
<ipython-input-19-fb82ca23d179> in <module>
     20 import cv2
     21 import tensorflow as tf
---> 22 import segmentation_models as sm
     23 model = tf.keras.applications.segmentation_model(model_name='mask_rcnn', weights='imagenet')
     24 get_ipython().run_line_magic('load_ext', 'tensorboard')

3 frames
/usr/local/lib/python3.8/dist-packages/efficientnet/__init__.py in init_keras_custom_objects()
     69     }
     70 
---> 71     tensorflow.python.keras.utils.generic_utils.get_custom_objects().update(custom_objects)
     72 
     73 

AttributeError: module 'keras.utils.generic_utils' has no attribute 'get_custom_objects'

库应成功导入。 我的张量流版本是 2.11.0 我正在使用 google colab

Answer 1

降级

tensorflow

到版本

1.x

或升级

segmentation_models

以便它可以与

tensorflow_2.x

一起工作。如果您不能同时执行这两项操作，请使用 TensorFlow 的

compat

行为，如下所示。

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()