import tensorflow as tf
tf_version = int(tf.__version__.split(".")[0])
if tf_version == 1:
from keras.models import Model, Sequential
from keras.layers import Input, Convolution2D, ZeroPadding2D, MaxPooling2D, Flatten, Dense, Dropout, Activation
else:
from tensorflow import keras
from tensorflow.keras.models import Model, Sequential
from tensorflow.keras.layers import Input, Convolution2D, ZeroPadding2D, MaxPooling2D, Flatten, Dense, Dropout, Activation
根据tensorflow版本修改导入包
最新推荐文章于 2022-10-08 12:01:30 发布