pix2pixHD安装

Posted 2022-11-30 明天去哪

安装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等进行剪切