代码拉取完成,页面将自动刷新
将处理好的数据集放到模型目录的 /data 文件夹下
安装依赖
pip3.7 install ./requirements/requirements.txt
注:pillow建议安装较新版本, 与之对应的torchvision版本如果无法直接安装,可使用源码安装对应的版本,源码参考链接:https://github.com/pytorch/vision ,建议Pillow版本是9.1.0 torchvision版本是0.6.0
安装mmcv
git clone -b 1.3.9 https://github.com/open-mmlab/mmcv.git
cd mmcv
MMCV_WITH_OPS=1 pip install -e . # 安装full版本
更改底层文件,分别修改mmcv和apex的底层源码
# mmcv_need
/bin/cp -f additional_need/mmcv/distributed.py ../mmcv/mmcv/parallel/
/bin/cp -f additional_need/mmcv/dist_utils.py ../mmcv/mmcv/runner/
/bin/cp -f additional_need/mmcv/optimizer.py ../mmcv/mmcv/runner/hooks/
/bin/cp -f additional_need/mmcv/text.py ../mmcv/mmcv/runner/hooks/logger/
# apex_need
/bin/cp -f additional_need/amp/scaler.py ../apex/amp/
# 精度
sh ./test/run_1p.sh
# 性能
sh ./test/run_1p_perf.sh
chmod a+x ./tools/dist_train.sh
# 精度
sh ./test/run_8p.sh
# 性能
sh ./test/run_8p_perf.sh
运行产生的log文件和模型的参数文件在 项目目录的 work_dirs 文件夹下
代码涉及公网地址参考 public_address_statement.md
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。