【Windows11 安装 Detectron2】

Windows11 安装 Detectron2

  • [1. 创建虚拟环境](#1. 创建虚拟环境)
  • 2.配置Pytorch环境
  • [3. 安装cocoapi](#3. 安装cocoapi)
  • [4. 下载detectron2](#4. 下载detectron2)
    • [4.1 修改setup.py](#4.1 修改setup.py)
    • [4.2 修改 nms_rotated_cuda.cu(detectron2\detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu)](#4.2 修改 nms_rotated_cuda.cu(detectron2\detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu))
  • [5. 开始下载依赖库以及编译环境](#5. 开始下载依赖库以及编译环境)
  • 6.测试
  • [7. demo代码](#7. demo代码)
  • [8. 参考文章](#8. 参考文章)

1. 创建虚拟环境

bash 复制代码
conda create -n detectron1 python=3.8
# 激活环境
conda activate detectron1

2.配置Pytorch环境

bash 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

3. 安装cocoapi

bash 复制代码
pip install pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple

4. 下载detectron2

bash 复制代码
git clone https://gitcode.com/facebookresearch/detectron2.git

4.1 修改setup.py

bash 复制代码
将"pycocotools>=2.0.2", 修改为"pycocotools"

4.2 修改 nms_rotated_cuda.cu(detectron2\detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu)

将using namespace detectron2之前的用下面代码覆盖

cpp 复制代码
#include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h>
#include <ATen/cuda/CUDAApplyUtils.cuh>
/*#ifdef WITH_CUDA
#include "../box_iou_rotated/box_iou_rotated_utils.h"
#endif
// TODO avoid this when pytorch supports "same directory" hipification
#ifdef WITH_HIP
#include "box_iou_rotated/box_iou_rotated_utils.h"
#endif*/
#include "box_iou_rotated/box_iou_rotated_utils.h"

5. 开始下载依赖库以及编译环境

命令要在这个目录下执行,不要进入到detectron代码中去:

执行如下命令:

cpp 复制代码
python -m pip install -e detectron2 -i https://pypi.tuna.tsinghua.edu.cn/simple

等待编译完成即可。

6.测试

cpp 复制代码
Study\ImageSegmentation\detectron2\demo>python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input 000043.jpg input2.jpg  --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

7. demo代码

python 复制代码
import wget
import cv2

import torch
import detectron2
from detectron2.utils.logger import setup_logger
setup_logger()

# import some common libraries
import numpy as np
import os, json, cv2, random

# import some common detectron2 utilities
from detectron2 import model_zoo
from detectron2.engine import DefaultPredictor
from detectron2.config import get_cfg
from detectron2.utils.visualizer import Visualizer
from detectron2.data import MetadataCatalog, DatasetCatalog


TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])
CUDA_VERSION = torch.__version__.split("+")[-1]
print("torch: ", TORCH_VERSION, "; cuda: ", CUDA_VERSION)
print("detectron2:", detectron2.__version__)

def progress_bar(current, total, width=80):
    progress = current / total
    bar = '#' * int(progress * width)
    percentage = round(progress * 100, 2)
    print(f'[{bar:<{width}}] {percentage}%')
    

if True:
    url = 'http://images.cocodataset.org/val2017/000000439715.jpg'
    save_path = 'input.jpg'

    try:
        wget.download(url, save_path, bar=progress_bar)
    except Exception as e:
        print(f'An error occurred: {e}')

im = cv2.imread("./input.jpg")
cv2.imshow('img', im)
cv2.waitKey(0)

cfg = get_cfg()
# add project-specific config (e.g., TensorMask) here if you're not running a model in detectron2's core library
cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5  # set threshold for this model
# Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as well
cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
predictor = DefaultPredictor(cfg)
outputs = predictor(im)

# look at the outputs. See https://detectron2.readthedocs.io/tutorials/models.html#model-output-format for specification
print(outputs["instances"].pred_classes)
print(outputs["instances"].pred_boxes)

# We can use `Visualizer` to draw the predictions on the image.
v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
cv2.imshow('results', out.get_image()[:, :, ::-1])
cv2.waitKey(0)

8. 参考文章

https://blog.csdn.net/weixin_45839733/article/details/129356470

相关推荐
我爱一条柴ya12 分钟前
【AI大模型】神经网络反向传播:核心原理与完整实现
人工智能·深度学习·神经网络·ai·ai编程
万米商云16 分钟前
企业物资集采平台解决方案:跨地域、多仓库、百部门——大型企业如何用一套系统管好百万级物资?
大数据·运维·人工智能
新加坡内哥谈技术19 分钟前
Google AI 刚刚开源 MCP 数据库工具箱,让 AI 代理安全高效地查询数据库
人工智能
慕婉030721 分钟前
深度学习概述
人工智能·深度学习
大模型真好玩22 分钟前
准确率飙升!GraphRAG如何利用知识图谱提升RAG答案质量(额外篇)——大规模文本数据下GraphRAG实战
人工智能·python·mcp
198923 分钟前
【零基础学AI】第30讲:生成对抗网络(GAN)实战 - 手写数字生成
人工智能·python·深度学习·神经网络·机器学习·生成对抗网络·近邻算法
6confim23 分钟前
AI原生软件工程师
人工智能·ai编程·cursor
阿里云大数据AI技术24 分钟前
Flink Forward Asia 2025 主旨演讲精彩回顾
大数据·人工智能·flink
i小溪25 分钟前
在使用 Docker 时,如果容器挂载的数据目录(如 `/var/moments`)位于数据盘,只要服务没有读写,数据盘是否就不会被唤醒?
人工智能·docker
程序员NEO27 分钟前
Spring AI 对话记忆大揭秘:服务器重启,聊天记录不再丢失!
人工智能·后端