前面的步骤与以下链接1相同,其中使用conda配置pytorch环境时,参考pytorch官网链接2:
运行detectron2里的文件,报错为:
AttributeError: module 'PIL.Image' has no attribute 'LINEAR'
上网查询发现是pillow版本太高,为10.0.0, pillow貌似是安装torch或者torchvision的时候一并安装的,我当时是去找对应的.whl下载安装的,并不是用pytorch官网的指令。
但是用pip uninstall pillow 再指定9.4.0版本安装并没有解决问题,反而产生其他报错,猜测与我的torch、torchvision版本有关。
因此重新配置pytorch环境,具体链接如下:
# CUDA 11.3
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
查看pillow版本为9.4.0,未产生前面所述的错误。
具体使用时,还需安装scipy、opencv-python库,直接pip install 安装即可。