跑通lite hrnet
1.连接autoAL并激活conda
激活conda环境
①.# 查看系统所使用的 shell
echo $SHELL
②.# 初始化 conda
conda init bash && source /root/.bashrc
2.解压coco数据集
unzip val2017.zip -d /root/autodl-tmp/chen/Lite-HRNet-hrnet/data/coco
unzip train2017.zip -d /root/autodl-tmp/chen/Lite-HRNet-hrnet/data/coco
unzip annotations_trainval2017.zip -d /root/autodl-tmp/chen/Lite-HRNet-hrnet/data/coco
3.进去lite-hrnet文件
cd /root/autodl-tmp/chen/Lite-HRNet-hrnet
4.安装requirements
pip install -r requirement.txt
5.测试安装命令
跑通版:
python tools/test.py /root/autodl-tmp/chen/Lite-HRNet-hrnet/configs/top_down/lite_hrnet/coco/litehrnet_18_coco_256x192.py \
/root/autodl-tmp/chen/Lite-HRNet-hrnet/checkpoints/litehrnet18_coco_256x192-6bace359_20211230.pth \
--eval mAP
6.train命令
python tools/train.py configs/top_down/lite_hrnet/coco/litehrnet_18_coco_256x192.py
Q1运行结果出现
解决:import sys
sys.path.append("/root/autodl-tmp/chen/code/Lite-HRNet-hrnet")
Q2运行结果出现:
解决:
将models/backbones/litehrnet.py中的@BACKBONES.register_module()
改为@BACKBONES.register_module(force=True)
Q3运行结果出现:
这是因为在配置参数的时候没有加--的原因,解决:
Q4运行结果出现:
原因:文件缺失
解决:
从mmpose config 下复制个__base__(https://github.com/open-mmlab/mmpose/tree/master/configs/_base_/)粘贴到缺失的位置即可。
Q5运行结果出现:
解决:对应位置添加即可:
Q6运行结果出现:
Caught KeyError in DataLoader worker process 0.
KeyError: 'center'
解决:
根据查找结果,重新安装mmcv和mmpose,将mmcv版本改为1.6.0,没有出现问题了。
参考文章:
pycharm相关错误及解决(三)——Lite-HRNet - Thirteen13th - 博客园 (cnblogs.com)