xinet.ModuleTool

class xinet.ModuleTool(inputs: torch.Tensor, channel: str = 'first', mean: list[float] = [0.485, 0.456, 0.406], std: list[float] = [0.229, 0.224, 0.225])

将 inputs 转换为 NumPy 格式

参数

• inputs – 批量数据

• mean – 默认为 ImageNet 的 mean

• std – 默认为 ImageNet 的 std

• channel – 取值范围为 [‘first’, ‘last’]

上一页

xinet

下一页

xinet.CV