Mask DINO环境配置

本文详细介绍了如何在CUDA11.3环境中配置PyTorch1.10.1、torchvision、torchaudio,以及如何安装和使用Detectron2和MaskDINO库进行实例分割的步骤,包括代码示例和测试过程。
摘要由CSDN通过智能技术生成

1. 准备工作

cuda11.3
cudnncuda11对应最新版即可
pytorch1.10.1
torchvision0.11.2
torchaudio0.10.1

2. 环境

conda create -n env_name python=3.9
conda activate env_name
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

#安装detectron2
git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2


#克隆maskdino库
git clone https://github.com/IDEA-Research/MaskDINO.git
cd ./MaskDINO
pip install -r requirement.txt
cd maskdino/modeling/pixel_decoder/ops
python setup.py build install

 3.运行demo

将图片和模型分别放在imgs和weights中,进行测试

python .\demo\demo.py --config-file .\configs\coco\instance-segmentation\maskdino_R50_bs16_50ep_3s.yaml --input .\img\1.jpg --output .\img\2.jpg --opts MODEL.WEIGHTS .\weights\maskdino_r50_50ep_300q_hid1024_3sd1_instance_maskenhanced_mask46.1ap_box51.5ap.pth

OK,顺利结束

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值