pytorch如何加载标签在图片名上的数据集

1、pytorch如何加载标签在图片名上的数据集

2、首先是加载数据过程中需要调用的库

3、然后继承dataset类进行数据集的加载

4、首先获取数据集的地址列表

5、然后根据获取的数据

6、通过对文件名进行提取得到标签并将标签进行数字化

7、最后通过len得到数据集的长度，直接进行调用即可加载数据集