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维度设置的数值大一点就是绿色了
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维度设置的数值大一点就是绿色了