环境

PyTorch安装教程：Windows | Linux | Mac

代码

from torchvision.transforms import ToPILImage
import torch


def random_noise(nc, width, height):
    '''Generator a random noise image from tensor.

    If nc is 1, the Grayscale image will be created.
    If nc is 3, the RGB image will be generated.

    Args:
        nc (int): (1 or 3) number of channels.
        width (int): width of output image.
        height (int): height of output image.
    Returns:
        PIL Image.
    '''
    img = torch.rand(nc, width, height)
    img = ToPILImage()(img)
    return img


if __name__ == '__main__':
    random_noise(3, 256, 256).save('random-noise.jpg')

结果

彩色图	灰度图

参考

https://pytorch.org/docs/stable/generated/torch.rand.html#torch-rand
https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.ToPILImage

最后

以上就是风中春天为你收集整理的【PyTorch】生成一张随机噪声图片环境代码结果参考的全部内容，希望文章能够帮你解决【PyTorch】生成一张随机噪声图片环境代码结果参考所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。