图像增强
autoalbument的Python项目详细描述
自动唱片
AutoAlbument是一个AutoML工具,它使用Faster AutoAugment algorithm从数据中学习图像增强策略。它减轻了用户手动选择增强和调整其参数的负担。AutoAlbument为扩充管道提供了一个完整的随时可用的配置。在
该库支持图像分类和语义分割任务。您可以使用Albumentations在计算机视觉管道中利用AutoAlbument发现的策略。在
文档可从https://albumentations.ai/docs/autoalbument/获得
安装
AutoAlbument需要Python3.6或更高版本。要从PyPI安装最新的稳定版本:
pip install -U autoalbument
如何使用自动相册
- 您需要创建一个包含AutoAlbument参数的配置文件和一个为您的数据实现自定义PyTorch数据集的Python文件。接下来,您需要将这些文件传递到AutoAlbument。在
- AutoAlbument将使用生成性对抗性网络来发现扩充策略,然后创建一个包含这些策略的文件。在
- 最后,您可以使用Albumentations从文件加载扩充策略,并在计算机视觉管道中使用它们。在
您可以在https://albumentations.ai/docs/autoalbument/how_to_use/处阅读所有步骤的详细说明
示例
^{
要使用示例配置运行搜索:
autoalbument-search --config-dir </path/to/directory_with_dataset.py_and_search.yaml>
- 项目
标签: