复现论文《Localization with Sampling-Argmax》-未完结

论文链接：https://jeffli.site/sampling-argmax/resources/neurips2021-sampling-argmax.pdf

开源地址：https://github.com/Jeff-sjtu/sampling-argmax

1.安装环境

1.1安装Install Pytorch

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

由于我之前装过pytorch，跳过这一步。

1.2安装sampling_argmax

python setup.py develop

2.获取数据

一共三种数据集，包含MSCOCO, Human3.6M and MTFL，由于之前下载过coco数据集，所以只测试coco，把下载好的数据集按照以下形式放置。

 

3.从0开始训练

sudo python ./scripts/train_pose.py --nThreads 12 --exp-id coco_samp --cfg configs/coco/256x192_res50_lr1e-3_1x-simple-integral.yaml

 

训练过程中遇到的问题

(1)在配置文件256x192_res50_lr1e-3_1x-simple-integral.yaml中有一条语句为DET_FILE: './exp/json/test_det_rcnn.json'，但是作者并没有提供这个文件。

解决方案：我从作者的其他的github仓库找到了这个文件，链接为https://github.com/Jeff-sjtu/res-loglikelihood-regression/blob/master/exp/json/test_det_rcnn.json，下载之后放到指定文件夹即可。

(2)训练过程中出现找不到设备，通过在终端输入nvidia-smi报Failed to initialize NVML: Driver/library version mismatch

解决方案：重启电脑

目前还在训练中，待更新