Separate the image channels into r,g,b using OpenCV, ... pass tuple of axis you want results for in dim para of torch.mean and torch.std .
確定! 回上一頁