替换数据集ImageFolder(Pytorch)与RVL-CDIP数据集一起使用

时间:2020-06-17 12:36:25

标签: python image-processing deep-learning pytorch

我目前正在使用带有Pytorch的深度学习进行文档识别项目。

我正在使用torchvision.datasets中的ImageFolder来指示包含我的图像的文件夹的路径并对其进行规范化,但是有人告诉我,我将不得不使用RVL-CDIP数据集来训练我的模型,类的目录树。

我可以使用什么替代品来处理该数据集?

0 个答案:

没有答案
相关问题