1. 提供主流的model,和常用数据集
2. 对 torch.utils.data.Dataset进行扩充,针对不同类别图像放入不同文件夹的数据进行读取, torchvision.datasets.ImageFolder是torch.utils.data.Dataset 的子类。
3.提供transforms
1. 提供主流的model,和常用数据集
2. 对 torch.utils.data.Dataset进行扩充,针对不同类别图像放入不同文件夹的数据进行读取, torchvision.datasets.ImageFolder是torch.utils.data.Dataset 的子类。
3.提供transforms
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。