深度学习之:ImageDataGenerator的注意点

图片增强

注意点:ImageDataGenerator()增强扩充的是’original_data’里面子文件夹的图片,所以一定要有子文件夹,

from keras.preprocessing.image import ImageDataGenerator
path = 'original_data'#图片加强的文件路径
dst_path = 'gen_data'#加强后的图片存放的文件夹
datagen = ImageDataGenerator(rotation_range=10,width_shift_range=0.1
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
鸟类识别是一个常见的计算机视觉应用,可以使用深度学习技术来实现。以下是一个简单的鸟类识别代码示例,使用Keras框架和ResNet50模型。 ```python # 导入必要的库 from keras.preprocessing.image import ImageDataGenerator from keras.models import Sequential from keras.layers import Dense, Dropout, Flatten from keras.applications.resnet50 import ResNet50 from keras.optimizers import Adam # 设置参数 num_classes = 20 image_size = (224, 224) batch_size = 32 # 创建数据生成器 train_datagen = ImageDataGenerator( rescale=1./255, shear_range=0.2, zoom_range=0.2, horizontal_flip=True) test_datagen = ImageDataGenerator(rescale=1./255) train_generator = train_datagen.flow_from_directory( 'train/', target_size=image_size, batch_size=batch_size, class_mode='categorical') test_generator = test_datagen.flow_from_directory( 'test/', target_size=image_size, batch_size=batch_size, class_mode='categorical') # 构建模型 model = Sequential() model.add(ResNet50(include_top=False, pooling='avg', weights='imagenet')) model.add(Dense(512, activation='relu')) model.add(Dropout(0.5)) model.add(Dense(num_classes, activation='softmax')) # 编译模型 model.layers[0].trainable = False model.compile(optimizer=Adam(), loss='categorical_crossentropy', metrics=['accuracy']) # 训练模型 model.fit_generator( train_generator, steps_per_epoch=train_generator.samples // batch_size, validation_data=test_generator, validation_steps=test_generator.samples // batch_size, epochs=10) # 评估模型 scores = model.evaluate_generator(test_generator, steps=test_generator.samples // batch_size) print("Accuracy: {:.2f}%".format(scores[1]*100)) ``` 在上述代码中,我们首先导入了必要的库,并设置了一些参数。我们使用了Keras的ImageDataGenerator来创建数据生成器,并将图像进行缩放、裁剪、旋转等数据增强操作。我们使用了ResNet50模型作为基本模型,并添加了一些全连接层来进行分类。我们使用Adam优化器和交叉熵损失函数来编译模型。最后,我们使用fit_generator函数来训练模型,并使用evaluate_generator函数来评估模型的性能。 需要注意的是,我们在使用ResNet50模型时将其第一层冻结了,这是因为我们只想训练我们添加的全连接层,而不是整个模型。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

人工智能小李子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值