图像配置¶
enable_tensorflow_image¶
Enable Image Transformer for processing of image data (String)
默认值 'auto'
是否使用预训练的深度学习模型对图像数据进行处理,作为特征工程管道的一部分。图像(jpg、png 等)的 URI 列将通过 ImageNet 预训练的深度学习模型转换为数字表示。如未找到 GPU,则必须设置为”开启”以启用。
tensorflow_image_pretrained_models¶
Supported ImageNet pretrained architectures for Image Transformer (List)
默认值 ['xception']
图像转换器支持的 ImageNet 预训练架构。非默认架构需要网络连接以从 H2O S3 存储桶下载预训练的模型(要获取所有模型,下载 http://s3.amazonaws.com/artifacts.h2o.ai/releases/ai/h2o/pretrained/dai_image_models_1_10.zip 并在 tensorflow_image_pretrained_models_dir 中解压缩)。
tensorflow_image_vectorization_output_dimension¶
Dimensionality of feature space created by Image Transformer (List)
默认值 [100]
图像转换器创建的特征(嵌套)空间的维数。如果选择的数量大于 1,则多个转换器可以同时有效。
tensorflow_image_fine_tune¶
Enable fine-tuning of pretrained models used for Image Transformer (Boolean)
默认值 False
对用于图像转换器的 ImageNet 预训练模型进行微调。启用此功能会减慢训练速度,但应可提高准确度。
tensorflow_image_fine_tuning_num_epochs¶
Number of epochs for fine-tuning used for Image Transformer (Number)
默认值 2
对用于图像转换器的 ImageNet 预训练模型进行微调的时期数。
tensorflow_image_augmentations¶
List of augmentations for fine-tuning used for Image Transformer (List)
默认值 ['HorizontalFlip']
可能的图像增广列表,在对用于图像转换器的 ImageNet 预训练模型进行微调时需应用此列表。欲了解各个增强的详情,可访问 https://albumentations.ai/docs/。
tensorflow_image_batch_size¶
Batch size for Image Transformer. Automatic: -1 (Number)
默认值 -1
图像转换器的批处理大小。架构和批处理大小越大,内存使用越多。
tensorflow_image_pretrained_models_dir¶
Path to pretrained Image models. It is used to load the pretrained models if there is no Internet access. (String)
默认值 './pretrained/image/'
预训练的图像模型的路径。要获取所有模型,请下载 http://s3.amazonaws.com/artifacts.h2o.ai/releases/ai/h2o/pretrained/dai_image_models_1_10.zip,然后将其提取到文件夹并存储在安装了 Driverless AI 的实例上的目录中。
image_download_timeout¶
Image download timeout in seconds (Number)
默认值 60
如果图像由 URL 提供,等待图像下载的最大秒数
string_col_as_image_max_missing_fraction¶
Max allowed fraction of missing values for image column (Float)
默认值 0.1
字符串列中缺失元素的最大分数,以将其视为潜在的图像路径 (URI)
string_col_as_image_min_valid_types_fraction¶
Min. fraction of images that need to be of valid types for image column to be used (Float)
默认值 0.8
需具有有效结尾(由 string_col_as_image_valid_types 定义)的(唯一)图像 URI 的分数,以使字符串列被视为图像数据
tensorflow_image_use_gpu¶
Enable GPU(s) for faster transformations of Image Transformer. (Boolean)
默认值 True
是否通过图像转换器使用 GPU(如可用)将图像转换为嵌套。可能会大大加快速度。
max_image_auto_ddp_cores¶
Maximum number of cores to use for image auto model parallel data management (Number)
默认值 10
控制要用于图像自动模型并行数据管理的最大核心数。若设置为 0,则禁用 mp: https://pytorch-lightning.readthedocs.io/en/latest/guides/speed.html