参考自b站up林亦LYi的开源项目
传送门
本次只训练了cnn版本的
第一次接触这种项目,建python虚拟环境时出了点难以说清楚的小问题,安装不上requirement.txt中的gym库那个版本,折腾了一会,自己都乱了头绪,最后导致训练用的是gymnasium这个库,测试用的是gym库。
如上图中带"_test"的py文件就是我自己魔改时加上去的,用于测试。另一个则是训练。
将之前的一个个测试改成了检测文件夹,批量测试。注意,如果同时训练和测试,可能会爆内存。
test_cnn_result.txt记录测试结果
参数可能要自己改一下。不知为啥,一样的代码,一样的参数,就是跑不出视频里的效果,跑出来一个怂怂的货。改了以后才正常一点。
记录一下
步数不是越大越好,不要无脑将步数小的删掉
看结果时仔细看最高分,差点就把最厉害的删掉了
链接
链接: https://pan.baidu.com/s/1dsmpg43iGeATeKqpfwtcMg?pwd=6666 提取码: 6666)