版权声明：新建了专注于语义分割的QQ群704803384，欢迎交流!!! https://blog.csdn.net/u014451076/article/details/80578461
安装pytorch
这个照着官网来就行，本人使用pytorch 0.4.0 cuda 9.0，可以进行使用
sudo pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp27-cp27mu-linux_x86_64.whl
sudo pip install torchvision安装dominate
sudo pip install dominate
为了图像在显示的时候生成html
安装tensorrt
先在官网下载tensort 4.0，这里必须使用4.0，因为pix2pixHD中有的包旧版的tensorrt不存在，地址https://developer.nvidia.com/nvidia-tensorrt-4x-download
解压，添加环境变量
vim ~/.zshrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/path-to/TensorRT-4/lib
source ~/.zshrc设置路径
sudo vim /etc/profile.d/cuda.sh
export PATH=/usr/local/cuda-9.0/bin:$PATH
sudo su -
sudo pip install tensorrt-*.whl
exit安装uff
sudo pip install uff*.whl需要验证的话，在python中导入uff和tensorrt试一下
Clone
git clone https://github.com/NVIDIA/pix2pixHD
测试
cd pix2pixHD
./scripts/test_1024p.sh这个地方需要注意一下，如果要测试1024x2048的图像似乎很难，查看test_1024p.sh脚本可以发现
python test.py --name label2city_1024p --netG local --ngf 32 --resize_or_crop none其中resize_or_crop参数可以进行调节，使用resize则会将1024x2048图片resize到1024x1024的结果；使用crop需要设置参数–fineSize 512等进行剪切