windows下yolact++(保姆级教程,亲测!)

网上有不少教程,但实际走过一般发现全是坑,历时几天终于解决各种问题,在此记录。

首先是cuda版本问题,网上不少教程说,原环境是啥咱安啥,emmm,对也不全对,主要看自己硬件配置是啥,我的显卡是GeForce RXT3050,是安培架构,不支持cuda10.x系列,官方证据如下:(想自己查也可以,这是网址

 所以我没按照原版教程走,自己用的cuda11.1.

下一个主要的问题是DCNv2的安装编译,所有教程教的都很简单,可实际操作问题不断,我也找到了正确的解决方法和资源,可以参考我的这一篇文章。

解决在windows上DCNv2编译问题(强推!亲测有效!!)_Ningmoua的博客-CSDN博客

以下是完整步骤:

1、安装anaconda

2、下载源码

github下载源码及权重文件和准备测试图片(测试图片命名为test.jpg)。或者直接下载我的,下面是链接,0积分下载,用爱发电,请多支持!!

3、创建虚拟环境yolact

conda create -n yolact python=3.7
activate yolact

4、安装torch

需要看自己cuda版本,我用的cuda11.1,所以选择torch1.9(注意conda安装时是否变成了cpu版本,我踩过坑)

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

5、安装需要的依赖库

pip install cython opencv-python pillow  matplotlib 

6、安装cocoapi

这里推荐这篇文章的方法,屡试不爽。Win10下安装pycocotools_qq_29750461的博客-CSDN博客

或者输入

pip install pycocotools-windows

7、安装编译DCNv2

这是大坑,网上的方法很多种,大多没有效果,最后在github上找到方法,参考我另一篇文章。 解决在windows上DCNv2编译问题(强推!亲测有效!!)_Ningmoua的博客-CSDN博客

 8、测试

最后测试模型是否能跑通,在yolact文件夹下:

python eval.py --trained_model=weights/yolact_plus_resnet50_54_800000.pth --score_threshold=0.15 --top_k=15 --image=test.jpg

显示下面效果就说明没问题了:

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值