图像去雾：FSB-Dehazing-Framework，未成功

• 下载

https://github.com/superwsc/FSB-Dehazing-Framework

• 安装anaconda

https://blog.csdn.net/quantum7/article/details/156384942

• 安装相关

sudo apt-get install cmake build-essential libjpeg-dev libpng-dev conda create -n pytorch1 python=3.7 conda activate pytorch1 conda install pytorch=1.1 torchvision=0.3 cudatoolkit=9.0 -c pytorch pip install matplotlib scikit-image opencv-python yacs joblib natsort h5py tqdm pip install torch natsort opencv-python yacs

• 运行

INPUT_DIR=fog OUTPUT_DIR=defog python inference.py \ --input_dir ${INPUT_DIR} \ --result_dir ${OUTPUT_DIR} \ --save_images

• 报错

好像必须用cuda？