axes don't match array in pytorch

一個有意思的錯誤: self.transformer = transforms.Compose([transforms.ToTensor()]) img = self.transformer(img) 該語句在python3.7中運行無問題,在python2.7中出現如下錯誤: img = torch.from_numpy(pic.transpose((2, 0, 1))) ValueError:
相關文章
相關標籤/搜索