导入错误：无法从“pytorchcv”导入名称“load_mnist”

Question

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-1-2cacdf187bba> in <module>
      6 import numpy as np
      7 
----> 8 from pytorchcv import load_mnist, train, plot_results, plot_convolution, display_dataset
      9 load_mnist(batch_size=128)

ImportError: cannot import name 'load_mnist' from 'pytorchcv' (/anaconda/envs/py37_pytorch/lib/python3.7/site-packages/pytorchcv/__init__.py)

如何修复这个错误？ 我使用 python 3.7 和 Jupiter 笔记本。微软 Pytorch Fundamental 中的代码：链接 https://learn.microsoft.com/en-gb/learn/modules/intro-computer-vision-pytorch/5-convolutional-networks

import torch
import torch.nn as nn
import torchvision
import matplotlib.pyplot as plt
from torchinfo import summary
import numpy as np

from pytorchcv import load_mnist, train, plot_results, plot_convolution, display_dataset
load_mnist(batch_size=128)

我通过命令安装了PyTorch：pip install pytorchcv

Answer 1

我认为您可能有错误的 pytorchcv 包。 pypy中的不包含load_mnist

从头开始，您可以这样下载 mnist：

data_train = torchvision.datasets.MNIST('./data', download=True,train=True,transform=ToTensor()) data_test = torchvision.datasets.MNIST('./data', 下载=True，train=False，变换=ToTensor())

Answer 2

他们在导入 pytorchcv 之前错过了一个命令。 这个pytorchcv和PyPI中的pytorchcv不一样。

在导入 pytorchcv 之前运行此命令。

!wget https://raw.githubusercontent.com/MicrosoftDocs/pytorchfundamentals/main/computer-vision-pytorch/pytorchcv.py

然后它应该可以工作。

Answer 3

请从链接https://raw.githubusercontent.com/MicrosoftDocs/pytorchfundamentals/main/computer-vision-pytorch/pytorchcv.py下载准备好的py文件并将其放入当前文件夹，然后VS Code可以识别它

Answer 4

只需转到“.\Lib”并使用 https://raw.githubusercontent.com/MicrosoftDocs/pytorchfundamentals/main/computer-vision-pytorch/pytorchcv.py 更改“init.py”。