pytorch如何使用os库对数据集进行拆分

1、pytorch如何使用os库对数据集进行拆分？

2、首先导入需要的相关库

3、再设置需要拆分数据集的路径

4、然后设置拆分后的数据集的路径以及文件夹名

5、筛选符合要求的文件，如照片的jpg后缀

6、最后设置需要划分数据集的大小和个数即可