前言
先说收获,这此虽然耗时较长并且暂时还不算成功,但是这是我修改的比较多bug的一次,算是练习了debug吧。这份代码的主要问题是:路径问题和相关命令的残缺。
源码的主要问题
直接说碰到的几个点
- 首先是对源码的路径区分,这应该是比较简单的,因为之前熟悉了一下cd的使用
- 然后就是这此的标签和数据集的加载都是通过CSV文件进行的,标签的CSV没什么讲的。主要是加载数据集的CSV,里边保存了训练图片的路径信息和目标区域坐标以及目标标签。数据集中的路径作者是自己的绝对路径,需要改正。
- 并且源码没有写入XML文件的代码和写入数据集加载的CSV文件的代码,我给补上了。学到了如何用pandas创建表格并写入CSV。
- 另外,源码通过网络下载的预训练模型也没有指定路径和名字。
- 另外是我自己的问题,代码中经常有路径切换来切换去的,后续我完善一下。
- 最后就是训练的问题了,快三个小时的时间才训练1个epoch,一共15个…最后是自动中断了