问题描述
我正在寻找用于对非MNIST数字进行分类的pytorch的示例。经过数小时的搜索,看来这些算法对我不利。有人有很好的榜样吗?谢谢。
解决方法
我将其发布为答案,因为我没有评论的代表,
请查看Google街景数据集(SVHN)。就像MNIST一样,但是数据中存在更多的噪声。您的另一种选择是使用GAN并制作更多以前几乎不存在的图像。您也可以尝试使用非英语mnist数据集(尽管它偏离了最初的目标)。
使用pytorch链接到SVHN:https://github.com/potterhsu/SVHNClassifier-PyTorch
链接到原始SVHN:https://pytorch.org/docs/stable/torchvision/datasets.html#svhn
P.S。您也可以尝试自己制作数据集!这样做很有趣。