siamcar训练

siamcar
首先参考pysot训练进行配置
教程1
教程2
数据集:百度网盘里面
链接:https://pan.baidu.com/s/1AGkA5VakgNr-ZykUroj5xg
提取码:7rse
我目前只用了det进行了训练,百度网盘里面的det是已经被裁剪过的,所以直接拿来训练了。但是训练的时候会报错RuntimeError
然后查找资料,把原来config.ymal里的BATCH_SIZE的80换成了20,就不报错了,因为用的自己的笔记本训练了一晚上才训练15%。同时train的这块的根目录要这样配置:
siamcar训练_第1张图片
然后修改了一下config.py文件的读取数据路径:
siamcar训练_第2张图片
我修改了这个数值__C.DATASET.VIDEOS_PER_EPOCH = 1000,但是训练的时候会自动×4,也不知道为啥,就直接变成了4000.原来是600000,训练起来太费时间,等到真正最后训练时候再调这个数据。
siamcar训练_第3张图片
同时我把原来冻结10epoch的后开始解冻把resnet50的最后三个区块以供训练进行了修改,直接把config.yaml和config.py文件进行了修改,这样就不会解冻backbone模块,只对后面的回归和分类网络进行微调。
siamcar训练_第4张图片
siamcar训练_第5张图片
并且数据改小了,所以只需要很少时间训练了模型,当然此时的结果肯定没有原来的好,我目前只是尝试一下训练以及得到结果。
siamcar训练_第6张图片
也遇到了一些别的问题,就是自己笔记本显卡不足问题,然后把BATCH_SIZE变成了5,原来是80,hold不住。两个config文件要修改。

你可能感兴趣的:(人工智能,机器学习,深度学习)