Pytorch运行MCNN
运行代码:http://www.sohu.com/a/301202174_715776
1.首先克隆代码:git clone https://github.com/leeyeehoo/CSRNet-pytorch.git
2.将克隆好的文件放到Anaconda文件夹中(文件名为:CSRNet-pytorch)
3.在anacond文件夹下打开一个终端:运行jupyter notebook
4.jupyter notebook 打开后,找到CSRNet-pytorch文件,进去后运行model_dateset_ipynb
5.逐个运行,需要花费一定时间,可以得到shanghaitech的密度图了
6. 如果是用python3.需要改动以下内容:
1)model.py中,18行中到xrange改为range
2)19行中进行更改:
7.进入Anaconda3中CSRNet-pytorch文件夹,打开一个终端
8.**pytorch:conda activate pytorch
9.运行:python train.py part_A_train.json part_A_val.json 0 0
注意:
1)CSRNet-pytorch中到.json文件,里边到路径需要改成自己到文件路径:
如:/home/***/shanghaitech/part_A_final/test_data/images/IMG_73.jpg