成都信息工程大学
12.11 pytorch学习（dataset获取数据集方法）
墨衍会员 · AI 创作全网分发 墨衍智能分发 通过墨衍一键同步至各平台 可同步平台 CSDN微信公众号微博知乎掘金百家号博客园抖音小红书 你的文章也可以这样分发 墨衍会员 ¥399起/年，写一次发全网
我也要 关注 + 加入社区12.11 pytorch学习（dataset获取数据集方法）
前提准备:蚂蚁蜜蜂数据集程序：from torch.utils.data import Datasetimport osfrom PIL import Imageclass Mydata(Dataset): def __init__(self, root_dir, label_dir): self.root_dir = root_dir self.label_dir = label_dir self.path = os.path.join
