SOLOv2win11环境配置(基于Detectron2)
GPU3050,python3.8,pytorch1.9.1,cuda11.1,cudnn8.8.0
pytorch-gpu的安装找官网命令就可以了,很简单贴出网址
https://pytorch.org/get-started/previous-versions/
安装完pytorch的基础环境后,开始下边的步骤。其实都是进入到setup.py目录然后build一下就好了,都会自己装好,缺什么再自己装一下。主要还是后边运行报错,主打一个详细记录(小白)
首先,各种文件位置如下:
1、Cython安装
这个版本我没有特殊指定,它自动装的0.29.35
pip install Cython
2、pycocotools安装
可以用git克隆,但我的习惯是直接下载到本地,下载地址:https://github.com/philferriere/cocoapi
cd PythonAPI #进入setup.py文件目录
python setup.py build_ext --inplace
python setup.py build_ext install
pip install matplotlib # 这边单独拎出来是因为我网不好,前边安装失败了
用以下代码进行测试,不报错就是安装成功
import pycocotools
from pycocotools.coco import COCO
3、fvcore安装
下载地址:https://github.com/facebookresearch/fvcore
python setup.py build --force develop #记得进入setup.py目录
然后因为网不好?好多包都没装上,下边是我单独拎出来装的包,如果一次性安装成功,这一小块就不用看了。
pip install tabulate==0.9.0
pip install tqdm==4.65.0
pip install pyyaml==5.4.1
pip install yacs==0.1.8
pip install pywin32==226
这些包装完记得再重复一遍
python setup.py build --force develop
4、detectron2安装
下载地址:https://github.com/facebookresearch/detectron2
python setup.py build --force develop
5、下载AdelaiDet
下载地址:https://github.com/aim-uofa/AdelaiDet
python setup.py build --force develop
到这看AdelaiDet-master/demo/demo.py应该会显示cv2没有安装
pip install opencv-python
不会报红了,但其实还有包没装好。运行以下命令(位置全部根据自己的更改)
python demo/demo.py --config-file configs/SOLOv2/R50_3x.yaml --input input1.jpg --opts MODEL.WEIGHTS SOLOv2_R50_3x.pth
会报错找不到pth文件,下载SOLOv2_R50_3x.pth,直接放在AdelaiDet-master下即可
进入https://github.com/aim-uofa/AdelaiDet/blob/master/configs/SOLOv2/README.md点击model即可,或者想下载别的模型文件也都可以,在AdelaiDet的主页可以选择。
配置文件,测试图片,权重文件全部都可以找到后,开始解决报错
测试报错及处理
ModuleNotFoundError: No module named ‘pywintypes’
(1)卸载重装pywin32,pip uninstall pywin32; pip install pywin32。记得装原版本
ModuleNotFoundError: No module named ‘google.protobuf’
(2)用pip卸载重装protobuf,google不管用,用conda install protobuf
AttributeError: module ‘distutils’ has no attribute ‘version’
(3)降低setuptools版本,pip uninstall setuptools ;pip install setuptools==58.0.4
AttributeError: module ‘numpy’ has no attribute ‘bool’.
(4)定位到报错文件报错位置,将bool改为bool_
demo测试环境报错解决到这里就结束了,最终会有一张椅子的输出。后边训练自己的模型,注册数据集已经完成了等后边再记录,flag先立在这里。附上自己环境最终的各种包和版本,方便各位查漏补缺。
absl-py 1.4.0
adelaidet 0.2.0
antlr4-python3-runtime 4.8
appdirs 1.4.4
beautifulsoup4 4.12.2
black 21.4b2
blas 1.0
ca-certificates 2023.01.10
cachetools 5.3.1
certifi 2023.5.7
charset-normalizer 3.1.0
click 8.1.3
cloudpickle 2.2.1
contourpy 1.0.7
cycler 0.11.0
cython 0.29.35
detectron2 0.5
editdistance 0.6.2
fonttools 4.39.4
future 0.18.3
fvcore 0.1.5.post20220414
google 3.0.0
google-auth 2.19.1
google-auth-oauthlib 1.0.0
grpcio 1.55.0
hydra-core 1.1.0
idna 3.4
imageio 2.31.0
importlib-resources 5.12.0
intel-openmp 2023.1.0
iopath 0.1.8
kiwisolver 1.4.4
lazy-loader 0.2
libprotobuf 3.20.3
markdown 3.4.3
markupsafe 2.1.3
matplotlib 3.7.1
mkl 2023.1.0
mkl-service 2.4.0
mkl_fft 1.3.6
mkl_random 1.2.2
mypy-extensions 1.0.0
networkx 3.1
ninja 1.10.2
ninja-base 1.10.2
numpy 1.24.3
numpy-base 1.24.3
oauthlib 3.2.2
omegaconf 2.1.0
opencv-python 4.7.0.72
openssl 1.1.1t
pathspec 0.11.1
pillow 9.5.0
pip 23.1.2
polygon3 3.0.9.1
portalocker 2.7.0
protobuf 4.23.2
pyasn1 0.5.0
pyasn1-modules 0.3.0
pycocotools 2.0
pydot 1.4.2
pyparsing 3.1.0b2
python 3.8.0
python-dateutil 2.8.2
pywavelets 1.4.1
pyyaml 5.4.1
rapidfuzz 2.13.7
regex 2023.6.3
requests 2.31.0
requests-oauthlib 1.3.1
rsa 4.9
scikit-image 0.21.0
setuptools 58.0.4
shapely 2.0.1
soupsieve 2.4.1
sqlite 3.41.2
tabulate 0.9.0
tbb 2021.8.0
tensorboard 2.13.0
tensorboard-data-server 0.7.0
termcolor 2.3.0
tifffile 2023.4.12
toml 0.10.2
tomli 2.0.1
torch 1.9.1+cu111
torchaudio 0.9.1
torchvision 0.10.1+cu111
tqdm 4.65.0
typing-extensions 4.6.2
urllib3 2.0.2
vc 14.2
vs2015_runtime 14.27.29016
werkzeug 2.3.4
wheel 0.38.4
yacs 0.1.8
zlib 1.2.13