Detectron2 快速开始，使用 WebCam 测试

本文将引导快速使用 Detectron2 ，介绍用摄像头测试实时目标检测。

Detectron2: https://github.com/facebookresearch/detectron2

环境准备

基础环境

Ubuntu 18.04
- 系统安装，可见制作 USB 启动盘，及系统安装
Nvidia Driver
- 驱动安装，可见 Ubuntu 初始配置 - Nvidia 驱动
Anaconda Python
- 从这里下载 Linux 版本，并安装

Detectron2

安装，

# 创建 Python 虚拟环境

conda create -n detectron2 python=3.8 -y

conda activate detectron2

# 安装 PyTorch with CUDA

conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.2 -c pytorch -y

# 安装 Detectron2

git clone https://github.com/facebookresearch/detectron2.git

python -m pip install -e detectron2

# 安装 OpenCV ，捕获相机图像及显示

pip install opencv-python

检查，

$ python - <<EOF

import torch, torchvision

print(torch.__version__, torch.cuda.is_available())

import cv2 as cv

print(cv.__version__)

EOF

1.7.1 True

4.5.1

安装文档 - 官方

现有模型进行推断

从其 model zoo 选择一个感兴趣的模型进行推断。这里以 COCO R50-FPN 3x 训练的各类模型进行演示。

下载 model 进如下路径，

detectron2/models/

├── COCO-Detection

│   └── faster_rcnn_R_50_FPN_3x

│       └── 137849458

│           ├── metrics.json

│           └── model_final_280758.pkl

├── COCO-InstanceSegmentation

│   └── mask_rcnn_R_50_FPN_3x

│       └── 137849600

│           ├── metrics.json

│           └── model_final_f10217.pkl

├── COCO-Keypoints

│   └── keypoint_rcnn_R_50_FPN_3x

│       └── 137849621

│           ├── metrics.json

│           └── model_final_a6e10b.pkl

└── COCO-PanopticSegmentation

    └── panoptic_fpn_R_50_3x

        └── 139514569

            ├── metrics.json

            └── model_final_c10459.pkl

目标检测 - Faster R-CNN

执行，

cd detectron2/

mkdir -p _output

python demo/demo.py \

--config-file configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml \

--input ../data/bicycle.jpg \

--output _output/bicycle_COCO-Detection.jpg \

--confidence-threshold 0.5 \

--opts MODEL.WEIGHTS models/COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl

结果，

实例分割 - Mask R-CNN

执行，

python demo/demo.py \

--config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \

--input ../data/bicycle.jpg \

--output _output/bicycle_COCO-InstanceSegmentation.jpg \

--confidence-threshold 0.5 \

--opts MODEL.WEIGHTS models/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

结果，

姿态估计 - Keypoint R-CNN

执行，

python demo/demo.py \

--config-file configs/COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x.yaml \

--input ../data/bicycle.jpg \

--output _output/bicycle_COCO-Keypoints.jpg \

--confidence-threshold 0.5 \

--opts MODEL.WEIGHTS models/COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x/137849621/model_final_a6e10b.pkl

结果，

全景分割 - Panoptic FPN

执行，

python demo/demo.py \

--config-file configs/COCO-PanopticSegmentation/panoptic_fpn_R_50_3x.yaml \

--input ../data/bicycle.jpg \

--output _output/bicycle_COCO-PanopticSegmentation.jpg \

--confidence-threshold 0.5 \

--opts MODEL.WEIGHTS models/COCO-PanopticSegmentation/panoptic_fpn_R_50_3x/139514569/model_final_c10459.pkl

结果，

WebCam 摄像头使用

获取本机的 WebCam 列表，

$ ls /dev/video*

/dev/video0  /dev/video1  /dev/video2  /dev/video3

# 查看 WebCam 列表

#   如下：有 0, 2 两个 videos

#   - 第一个是 video ，第二个是 metadata

#   - 从 Linux Kernel 4.16 开始，增加的 metadata node

$ sudo apt install v4l-utils

$ v4l2-ctl --list-devices

HD Webcam: HD Webcam (usb-0000:00:14.0-13):

	/dev/video0

	/dev/video1

HD Pro Webcam C920 (usb-0000:00:14.0-4):

	/dev/video2

	/dev/video3

# 查看某 WebCam 支持的格式、分辨率、fps 信息

$ v4l2-ctl -d 2 --list-formats-ext

demo/demo.py 可修改期望打开的摄像头及其分辨率等，

elif args.webcam:

  cam = cv2.VideoCapture(2)

  cam.set(cv2.CAP_PROP_FRAME_WIDTH, 640)

  cam.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)

  cam.set(cv2.CAP_PROP_FPS, 30)

  print(f"wencam: {cam.get(cv2.CAP_PROP_FRAME_WIDTH)}x{cam.get(cv2.CAP_PROP_FRAME_HEIGHT)} {cam.get(cv2.CAP_PROP_FPS)}")

运行，

python demo/demo.py \

--config-file configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml \

--webcam \

--confidence-threshold 0.5 \

--opts MODEL.WEIGHTS models/COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl

效果，

GoCoding 个人实践的经验分享，可关注公众号！

Detectron2 快速开始，使用 WebCam 测试的更多相关文章

快速搭建appium自动测试环境
首先申明本文是基本于Python与Android来快速搭建Appium自动化测试环境: 主要分为以下几个步骤: 前提条件: 1)安装与配置python环境,打开 Python官网,找到"Do ...
如何快速掌握DDT数据驱动测试？
1.前言 (网盗概念^-^)相同的测试脚本使用不同的测试数据来执行,测试数据和测试行为完全分离, 这样的测试脚本设计模式称为数据驱动.(网盗结束)当我们测试某个网站的登录功能时,我们往往会使用不同的用 ...
使用docker快速搭建Permeate渗透测试系统实践
一.背景笔者最近在做一场Web安全培训,其中需要搭建一套安全测试环境:在挑选渗透测试系统的时候发现permeate渗透测试系统比较满足需求,便选择了此系统:为了简化这个步骤,笔者将系统直接封装到了d ...
IntelliJ IDEA快速自动生成Junit测试类
1.背景测试是保证代码必不可少的环节,自己构建测试方法太慢,并且命名也不规范,idea中提供了,一键构建测试结构的功能...废话不多说,直接写步骤 2.步骤 1.在需要做测试的类的当前窗口,直接按快 ...
Sqlite 快速批量插入数据测试
public static int insertDbBatch() { string sql = ""; SQLiteConnection conn = new SQLiteCon ...
python pandas 　使用列表快速创建数据用于测试
创建数据和行append数据 >>> df = pd.DataFrame([['AA', 1.00], ['Ks', 2.00]], columns=['name', 'age']) ...
Rspec: everyday-rspec实操：第9章快速编写测试，编写快速的测试。
Make it work, make it right, make it fast. 测试运行的时间.应用和测试组件的增长,速度会越来越慢,目标是保持代码的readable, maintainable ...
疫情期，如何用A/B测试快速迭代你的产品？
作者:友盟+数据科学家杨玉莲.陆子骏冠状病毒来袭牵动着每个人的心,但是病毒影响的不仅仅是我们的健康,也以极快的速度极深远地影响了整个移动互联网的发展.主流阵地原本在线下的需求,如医疗和生鲜电商,快 ...
移动应用开发测试工具Bugtags集成和使用教程
前段时间,有很多APP突然走红,最终却都是樱花一现.作为一个创业团队,突然爆红是非常难得的机会.然并卵,由于没有经过充分的测试,再加上用户的激增,APP闪退.服务器数据异常等问题就被暴露出来,用户的流 ...
Testin云测试平台初体验
这几天偶然接触到了一个叫做Testin的云测试平台,经过一番体验,感觉还是不错的,因为里面提供了大量的测试机型,可以针对Android手机的严重碎片化现象做出比较全面的测试,同时Testin的测试内容 ...

随机推荐

'ModelOptions' object has no attribute 'get_field_names
peewee安装时随意了点.装了2.8.0的. 倒回到2.6.0就好了. sudo pip uninstall peewee sudo pip install peewee==2.6.0
SSH框架优缺点
SSH框架优缺点开源是3个框架共有的优点 Struts2框架(MVC框架)的优点如下: 1) 实现了MVC模式,层次结构清晰,使程序员只需关注业务逻辑的实现: 2) 丰富的标签库,大大提高了开发 ...
Git CMD - push: Update remote refs along with associated objects
命令格式 git push [--all | --mirror | --tags] [--follow-tags] [--atomic] [-n | --dry-run] [--receive-pac ...
MFC圆角背景移动边角底色毛刺解决方案
CRect rc; Graphics graphics(pDC->m_hDC); GetClientRect(&rc); CRgn m_rgn; if (m_pBgImage) { gr ...
浅谈V8引擎中的垃圾回收机制
最近在看<深入浅出nodejs>关于V8垃圾回收机制的章节,转自:http://blog.segmentfault.com/skyinlayer/1190000000440270 这篇文章 ...
ORA-64379: Action cannot be performed on the tablespace assigned to FastStart while the feature is enabled
解决方法: 禁止IM FastStart exec DBMS_INMEMORY_ADMIN.FASTSTART_DISABLE();
谷歌浏览器javascript错误提示插件
JavaScript-Errors-Notifier_v2.1.7 下载地址安装方法: http://chromecj.com/utilities/2014-09/181.html 设置方式:
Linux实际常用命令
1.删除0字节文件 find -type f -size 0 -exec rm -rf {} \; 2.查看进程按内存从大到小排列 ps -e -o “%C : %p : %z : %a”|sor ...
Jmeter的使用简介及实例
一.安装及配置环境1.安装 java环境该软件需要java环境,安装jdk,在百度自行查找安装环境变量配置:变量名JAVA_HOME 值:jdk的安装路径 ...
python------模块定义、导入、优化 ------->re模块
该部分内容引用自:http://www.cnblogs.com/alex3714/articles/5161349.html 谢谢原博主的文章,只是自己学习,无其他使用. re模块常用正则表达式符号 ...