首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Tensorflow CNN:图像转换需要SciPy。安装SciPy

Tensorflow CNN:图像转换需要SciPy。安装SciPy
EN

Stack Overflow用户
提问于 2022-03-22 23:18:40
回答 1查看 299关注 0票数 1

我正在使用tensorflow在木星笔记本中训练CNN,并得到以下错误:ImportError: Image transformations require SciPy. Install SciPy.

我使用以下命令安装了the,并确认它已安装:

代码语言:javascript
运行
复制
!pip install scipy
import scipy

我有三个分类类,我将图像分类为,我的代码是:

代码语言:javascript
运行
复制
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

代码语言:javascript
运行
复制
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)
EN

回答 1

Stack Overflow用户

发布于 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之外编写了代码,

进口枕木和酒窝!!

代码语言:javascript
运行
复制
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

模特成功了。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/71580074

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档