参考博客:https://blog.csdn.net/weixin_45622994/article/details/107711120
一. yolact win10 环境配置出现的问题
1. 打开anaconda下prompt,创建一个虚拟环境:
conda create --name py37 python=3.7
conda activate py37
2. 安装以下包:
pip install cython
pip install opencv-python
pip install pillow
pip install pycocotools#用此方式:pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
pip install matplotlib
3. 安装pytorch与torchvision:
安装pytorch时需要先去查看自己的cuda版本
nvcc --version
,然后选择适配 的pytorch版本
pytorch官网:https://pytorch.org/
选择与自己相匹配的版本,这里显示是我安装的选择。
理论上直接复制下面的就行,但是网不好的话就不是很好下,容易失败,所以采取了离线下载,安装。
具体的过程:(适用于win版本)
先进清华源https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=D
下载对应pytorch版本,我的是Python3.7,需要PyTorch-gpu版本,cuda需要11.1,找到对应的bz2文件,如图
还要下载对应的torchvision===0.6.1,如图
下载好就在命令行进入你下载的路径目录里面安装,并输入下面代码进行离线安装。
conda install 对应的安装包文件路径名字
我的具体操作如下:
而且离线安装的时候应该是 conda install 这个安装包的整个路径,而不是只有安装包的名字,会报错找不到。
两个离线安装完后还要安装cudatoolkit=11.1
conda install cudatoolkit=11.1
检测pytorch安装结果,显示如下结果则成功:
python
import torch
print(torch.cuda.is_available())
ture
二. 测试yolact:
python eval.py --trained_model=weights/yolact_base_54_800000.pth --score_threshold=0.15 --top_k=15 --image=test.jpg
在测试yolac的过程中遇到了一些问题:
直接输入 python eval.py --trained_model=weights/yolact_base_54_800000.pth – score_threshold=0.15 --top_k=15 --image=test.jpg 命令一直报错:
后来想了一下换成了这条命令:
果然成功!
参考博客:
- https://pytorch.org/get-started/locally/
- https://blog.csdn.net/q2682589934/article/details/117367440
3.https://blog.csdn.net/weixin_45092662/article/details/107299953utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-4.essearch_pc_relevant&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-4.essearch_pc_relevant
4. https://www.freesion.com/article/7455937008/