pytorch生成绿色背景图片
import torch
from torchvision.utils import save_image
grean_bgr = torch.zeros(3, 512, 512)
grean_bgr[1, :, :] = 120
save_image(grean_bgr, "grean_bgr.jpg")
pytorch 的颜色通道是rgb ，
数据形状chw，
c的第1维度设置的数值大一点就是绿色了
本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
THE END
二维码