pytorch制作自己的数据集并训练

　　从别的博主那里收集来的，自己尝试着复现一下，做一个记录。

　　这是一个红绿灯单色识别的过程

　　一、首先，创建一个自己的数据集。

　　二、读取数据集，继承Dataset实现Mydataset子类

　　新建一个.py文件，复制并修改下面的代码

　　修改的位置已标注

　　三、训练部分

　　四、验证训练结果