利用AutoDL跑通lite_hrnet

科技前沿 • 2025-03-24 20:07 • 阅读 53

大家好，我是讯享网，很高兴认识大家。

跑通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_.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)

Python迭代DataLoader时出现TypeError: Caught TypeError in DataLoader worker process 0.错误。_dataloader不可迭代_小心丶的博客-CSDN博客

利用AutoDL跑通lite_hrnet

相关推荐