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

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

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

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

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

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

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