Python 接口说明

说明

  • 对于资料中注明的类、接口为公开的,可供用户调用。对于其他类,接口均为内部使用,不建议直接调用。如有特殊需要可以通过源码查看。
  • 引入多模态 SDK 时会显式设置环境变量 HF_DATASETS_OFFLINEHF_HUB_OFFLINE 为 1,即显式启用 huggingface 的离线模式,不会联网获取数据。

源码查看方式

可以通过以下示例代码查看源码所在安装地址,并进入安装地址查看源码文件。

import mm
print(mm.__file__)

打印的内容即为源码文件所在的地址。

目录

数据类型

DataType

数据类型枚举类

属性名 说明
DataType.INT8 int8 类型。
DataType.UINT8 uint8 类型。
DataType.FLOAT32 float32 类型。

TensorFormat

Tensor 数据排布格式枚举类

属性名 说明
TensorFormat.ND ND 格式。表示通用 N 维数组。
TensorFormat.NHWC NHWC 格式。表示张量排布为 Batch, Height, Width, Channel。
TensorFormat.NCHW NCHW 格式。表示张量排布为 Batch, Channel, Height, Width。

ImageFormat

ImageFormat 枚举类

属性名 说明
ImageFormat.RGB RGB 类型。
ImageFormat.BGR BGR 类型。
ImageFormat.RGB_PLANAR RGB_PLANAR 类型。
ImageFormat.BGR_PLANAR BGR_PLANAR 类型。

LogLevel

日志级别枚举类

属性名 说明
LogLevel.DEBUG 调试级别。
LogLevel.INFO 提示级别。
LogLevel.WARN 警告级别。
LogLevel.ERROR 错误级别。
LogLevel.FATAL 致命错误级别。

DeviceMode

操作运行的模式

属性名 说明
DeviceMode.CPU 当前操作运行在 CPU 模式下。

Interpolation

resize 操作中使用的插值算法

属性名 说明
Interpolation.BICUBIC 双立方插值算法。