AQATrack开源模型探索

配置环境按照readme说明书傻瓜式探索anaconda终端输入conda create -n aqatrack python3.8 conda activate aqatrack bash install.sh最后一行Windows系统可能会报错缺少whl之类的没关系咱们打开PyCharm建一个项目aqatrack路径选aqatrack主干文件的根目录选现有环境然后在终端运行python tracking/create_default_local_file.py --workspace_dir . --data_dir ./data --save_dir ./output记得把里面的点换成自己 的绝对路径python tracking/create_default_local_file.py --workspace_dir.--data_dir./data--save_dir./output注意啊这里是反斜杠在终端运行以上命令来设置该项目的路径有报错就是模块缺失缺什么咱们就补什么直到不报错即可以上环境就配置好啦撒花✿✿ヽ(°▽°)ノ✿遇到安装包缺失的情况就用pip文件内容里面出现的陌生词汇tensorboard-记录欠拟合、过拟合问题的现在修改一些地方的参数yaml文件是256是基础384是骨干每一轮采样率23行-添加数据集名称29行-更改采样的比例66行-训练轮数BATCH_SIZE:8[可以按偶数上下调比如6、10怎么样的情况就往上调了呢]74行-显卡数量-T_T工作站只有一个显卡80行-验证次数VAL_EPOCH_INTERVAL:20lib.train.data_specs里面添加分好类的数据集txt文件还没加。。。lib.train.dataset里面的__init__.py文件里面需要改加入引入数据集名字的类from .ootb import OOTB这里 的ootb需要是lib.train.dataset里面的一个ootb.py文件这个ootb.py可以让D师傅手搓一个lib.train下属的base_functions.py里面改同样在文件开头导入数据集类# 在现有导入语句后添加 from lib.train.dataset import OOTB, OOTB_lmdb # 假设存在这些类修改names2datasets函数在32行添加数据集名称在81行添加数据集支持代码如下# 添加 OOTB 数据集支持 if name OOTB: if settings.use_lmdb: print(Building OOTB from lmdb) # 假设 OOTB_lmdb 需要的参数 datasets.append(OOTB_lmdb(settings.env.ootb_lmdb_dir, splittrain, image_loaderimage_loader)) else: print(Building OOTB dataset) # 假设 OOTB 需要的参数数据目录和分割 datasets.append(OOTB(settings.env.ootb_dir, splittrain, image_loaderimage_loader))以上参数修改完成可以开始运行了。补充训练的代码修改1设置数据集路径在README文件里复制代码替换点在lib.train.admin下会生成一个local.py文件里会注明路径如果报错缺少模块缺什么补什么用pip阿里镜像源就可以比如pip install torch -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com2设置环境在lib.train.admin下的environment.py、、3dataset里面改两个1添加读取数据集的py文件2-init_文件中导入库、、4train下修改data_specs里的txt训练和测试文件不需要起始帧范围、、4yaml文件中的参数修改显卡数量Batch Size按2的幂次来训练从高到底的试16-8-4-2测试的修改test读取数据集的文件生成参考latsot的读取方式test.evaluation里面修改环境environmentselfdatasettestepoch46改数据集名字78epochxxx训练权重名字训练过程中train.trainers下的base_trainer.py里面96-98修改训练轮数保存最后10轮可以修改权重修改的时机out of memory