COLOR_BGR2RGB) feature.append(torch.tensor(jpg).permute(2,0,1)) labels.append(torch.tensor(png).permute(2,0,1)) return feature,labels.
確定! 回上一頁