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

时间:2021-05-08 10:02:02
【文件属性】:
文件名称:Keras-augmentation-layer:Keras层的实现,该层使用GPU执行图像增强
文件大小:12KB
文件格式:ZIP
更新时间:2021-05-08 10:02:02
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)

网友评论