Keras-augmentation-layer:Keras层的实现,该层使用GPU执行图像增强

时间:2024-05-28 03:48:42
【文件属性】:

文件名称:Keras-augmentation-layer:Keras层的实现,该层使用GPU执行图像增强

文件大小:12KB

文件格式:ZIP

更新时间:2024-05-28 03:48:42

tensorflow keras layer gpu-acceleration augmentation

Keras层的实现,该层使用GPU执行图像增强。 它可以为ImageDataGenerator或独立扩充图书馆是一个很好的和快速更换 , 等。 重要说明:它是原型版本,我认为可以在速度和可用性方面进行很多改进。 我真的很想在Keras和TF官方存储库中看到它。 随意使用拉取请求添加更多变换或参数。 要求 Python 3。*,Keras 2。*,tensorflow> = 1.13 安装 pip install keras-augm-layer 用法 您需要定义初始模型,然后选择一组转换并添加AugmLayer作为模型的第一层。 请参见下面的示例: from keras . applications . mobilenet import MobileNet , preprocess_input from keras . layers import Dense , Input from


【文件预览】:
Keras-augmentation-layer-master
----README.md(6KB)
----benchmarks()
--------benchmark_gpu_augm.py(5KB)
--------benchmark_cpu_albumentations.py(6KB)
----keras_augm_layer()
--------__init__.py(17KB)
----check_augmentations.py(3KB)

网友评论