pytorch的图片加载预处理代码

Question

有两个positive和nagetive图片文件夹，各两万张图片，各自选取前16000个凑成训练集，剩下的凑成测试集，已经做好卷积网络以及计算正确率和loss部分代码，怎么编写关于上述部分的读取数据的代码。也即如何修改下面图片中的代码。

 

 

Answer 1

参考一下：https://blog.csdn.net/u012223913/article/details/103672545