Skip to content

有些自训练模型需要用到mead跟std参数。 #186

@Lockey-J

Description

@Lockey-J

自定义模型的 mean/std
tensor = np.array(img, dtype=np.float32) / 255.0
tensor = (tensor.transpose(2, 0, 1) - meta['mean']) / meta['std']
return np.expand_dims(tensor, axis=0)有些是需要用到这样的归一化算法。可以增加这部分的算法么。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions