运行环境:cuda 9.2 、python 3.6 、torch 0.4.1
Installation部分
- 运行./make.sh时报错,权限不够
解决方法:
cd PPDM-master/src/lib/models/networks/DCNv2
chmod 777 make.sh
./make.sh
2.复制环境后使用pip时出错(pip对应的python路径不对)
解决办法:
which pip
/home/wll/anaconda3/envs/test/bin/pip
which python
/home/wll/anaconda3/envs/test/bin/python
vim /home/wll/anaconda3/envs/test/bin/pip
把#!后面的改成python的路径就好了
3.torch版本换成0.4.1版本
手动下载torch安装包网址:(torch.pypi里面找不到想要的版本)
https://download.pytorch.org/whl/torch_stable.html
Training部分
4.
运行
cd src
python main.py hoidet --batch_size 112 --master_batch 7 --lr 4.5e-4 --gpus 0,1,2,3,4,5,6,7 --num_workers 16 --load_model ../models/ctdet_coco_dla_2x.pth --image_dir images/train2015 --dataset hico --exp_id hoidet_hico_dla
出现如下错误(加载模型时出错)
解决办法:将/home/wanglulu/PPDM_master/src/lib/models/networks/pose_dia_dcn.py 中的pretained的True改为False
5.如何解决RuntimeError: CUDA error: out of memory?(爆显存问题)
将batch size 改小,一般设置为2的幂次,32,64,128,…