我正在使用tensorflow在木星笔记本中训练CNN,并得到以下错误:ImportError: Image transformations require SciPy. Install SciPy.
我使用以下命令安装了the,并确认它已安装:
!pip install scipy
import scipy
我有三个分类类,我将图像分类为,我的代码是:
print(len(os.listdir('C:/images/cvd/Training/class1')))
print(len(os.listdir('C:/images/cvd/Testing/class1')))
print(len(os.listdir('C:/images/cvd/Training/class2')))
print(len(os.listdir('C:/images/cvd/Testing/class2')))
print(len(os.listdir('C:/images/cvd/Training/class3')))
print(len(os.listdir('C:/images/cvd/Testing/class3')))
1400 156 1022 114 1403 156
model = tf.keras.models.Sequential([
tf.keras.layers.Conv2D(64, (3, 3), activation='relu', input_shape=(80, 80, 1)),
tf.keras.layers.MaxPooling2D(2, 2),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Conv2D(64, (3, 3), activation='relu'),
tf.keras.layers.MaxPooling2D(2, 2),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(3, activation='softmax')
])
# compile model
model.compile(loss='sparse_categorical_crossentropy',
optimizer='adam',
metrics=['accuracy'])
# pre process images
TRAINING_DIR = 'C:/images/cvd/Training/'
train_datagen = ImageDataGenerator(rescale=1.0 / 255)
train_generator = train_datagen.flow_from_directory(TRAINING_DIR,
target_size=(80, 80),
batch_size=24,
class_mode='categorical')
VALIDATION_DIR = 'C:/images/cvd/Testing/'
validation_datagen = ImageDataGenerator(rescale=1.0 / 255)
validation_generator = validation_datagen.flow_from_directory(VALIDATION_DIR,
target_size=(80, 80),
batch_size=24,
class_mode='categorical')
history = model.fit(train_generator, epochs=20, steps_per_epoch=10,
validation_data=validation_generator, validation_steps=4)
发布于 2022-03-30 04:20:39
我也有同样的问题。但我通过以下方法解决了问题。首先,我的库版本*numpy : 1.21.5 *scipy : 1.7.3 *tensorflow-gpu : 2.8.0
我移到了错误行..。在我的例子中,affine_transformation.py可能错误行为20 ~22,我检查了是否已安装,是否存在错误。所以,我在try~catch之外编写了代码,
进口枕木和酒窝!!
import scipy
from scipy import ndimage
try:
import scipy
print(scipy.__version__)
#scipy.ndimage cannot be accessed until explicitly imported
print(ndimage.affine_transform.__doc__)
from scipy import ndimage
except ImportError as e:
print('test............',e)
scipy = None
模特成功了。
https://stackoverflow.com/questions/71580074
复制相似问题