ubuntu@fadfd17f190d:~/user_space/SGAN$ CUDA_VISIBLE_DEVICES=0 python tools/stage2/train_sgan.py --cfg_file config/sgan_vgg16_321x321.yml
Traceback (most recent call last):
File "tools/stage2/train_sgan.py", line 11, in <module>
from lib.dataset.get_dataset import get_dataset
File "/home/ubuntu/user_space/SGAN/tools/stage2/../../lib/dataset/get_dataset.py", line 4, in <module>
from lib.dataset import voc12_sgan
File "/home/ubuntu/user_space/SGAN/tools/stage2/../../lib/dataset/voc12_sgan.py", line 15, in <module>
from lib.dataset import imutils
File "/home/ubuntu/user_space/SGAN/tools/stage2/../../lib/dataset/imutils.py", line 11, in <module>
import krahenbuhl2013
File "/home/ubuntu/MyFiles/.local/lib/python3.5/site-packages/krahenbuhl2013/__init__.py", line 1, in <module>
from .CRF import *
File "/home/ubuntu/MyFiles/.local/lib/python3.5/site-packages/krahenbuhl2013/CRF.py", line 1, in <module>
from krahenbuhl2013.wrapper import DenseCRF
ImportError: No module named 'krahenbuhl2013.wrapper'
Cython版本问题;原来系统安装了Cython==0.29.1;但是CRF要求。所以要换版本
pip install Cython==0.24.1
然后,重新编译CRF,切记:不要乱改代码内容。