每次重装环境后需要修改文件加载部分,都要重新阅读一遍Pytorch源码。记录一下Pytorch加载数据源码的路径:
site-packages/torchvision/datasets/folder.py
其中在DatasetFolder类中定义了文件的加载方法,修改其中__getitem__函数即可。
每次重装环境后需要修改文件加载部分,都要重新阅读一遍Pytorch源码。记录一下Pytorch加载数据源码的路径:
site-packages/torchvision/datasets/folder.py
其中在DatasetFolder类中定义了文件的加载方法,修改其中__getitem__函数即可。