OpenCV实现机器人对物体进行移动跟随的方法实例
作者:Arcann 发布时间:2023-04-20 08:50:49
标签:opencv,机器人,移动
1.物体识别
本案例实现对特殊颜色物体的识别,并实现根据物体位置的改变进行控制跟随。
import cv2 as cv
# 定义结构元素
kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))
# print kernel
capture = cv.VideoCapture(0)
print capture.isOpened()
ok, frame = capture.read()
lower_b = (65, 43, 46)
upper_b = (110, 255, 255)
height, width = frame.shape[0:2]
screen_center = width / 2
offset = 50
while ok:
# 将图像转成HSV颜色空间
hsv_frame = cv.cvtColor(frame, cv.COLOR_BGR2HSV)
# 基于颜色的物体提取
mask = cv.inRange(hsv_frame, lower_b, upper_b)
mask2 = cv.morphologyEx(mask, cv.MORPH_OPEN, kernel)
mask3 = cv.morphologyEx(mask2, cv.MORPH_CLOSE, kernel)
# 找出面积最大的区域
_, contours, _ = cv.findContours(mask3, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)
maxArea = 0
maxIndex = 0
for i, c in enumerate(contours):
area = cv.contourArea(c)
if area > maxArea:
maxArea = area
maxIndex = i
# 绘制
cv.drawContours(frame, contours, maxIndex, (255, 255, 0), 2)
# 获取外切矩形
x, y, w, h = cv.boundingRect(contours[maxIndex])
cv.rectangle(frame, (x, y), (x+w, y+h), (255, 0, 0), 2)
# 获取中心像素点
center_x = int(x + w/2)
center_y = int(y + h/2)
cv.circle(frame, (center_x, center_y), 5, (0, 0, 255), -1)
# 简单的打印反馈数据,之后补充运动控制
if center_x < screen_center - offset:
print "turn left"
elif screen_center - offset <= center_x <= screen_center + offset:
print "keep"
elif center_x > screen_center + offset:
print "turn right"
cv.imshow("mask4", mask3)
cv.imshow("frame", frame)
cv.waitKey(1)
ok, frame = capture.read()
实际效果图
2.移动跟随
结合ROS控制turtlebot3或其他机器人运动,turtlebot3机器人的教程见我另一个博文:ROS控制Turtlebot3
首先启动turtlebot3,如下代码可以放在机器人的树莓派中,将相机插在USB口即可
代码示例:
import rospy
import cv2 as cv
from geometry_msgs.msg import Twist
def shutdown():
twist = Twist()
twist.linear.x = 0
twist.angular.z = 0
cmd_vel_Publisher.publish(twist)
print "stop"
if __name__ == '__main__':
rospy.init_node("follow_node")
rospy.on_shutdown(shutdown)
rate = rospy.Rate(100)
cmd_vel_Publisher = rospy.Publisher("/cmd_vel", Twist, queue_size=1)
# 定义结构元素
kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))
# print kernel
capture = cv.VideoCapture(0)
print capture.isOpened()
ok, frame = capture.read()
lower_b = (65, 43, 46)
upper_b = (110, 255, 255)
height, width = frame.shape[0:2]
screen_center = width / 2
offset = 50
while not rospy.is_shutdown():
# 将图像转成HSV颜色空间
hsv_frame = cv.cvtColor(frame, cv.COLOR_BGR2HSV)
# 基于颜色的物体提取
mask = cv.inRange(hsv_frame, lower_b, upper_b)
mask2 = cv.morphologyEx(mask, cv.MORPH_OPEN, kernel)
mask3 = cv.morphologyEx(mask2, cv.MORPH_CLOSE, kernel)
# 找出面积最大的区域
_, contours, _ = cv.findContours(mask3, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)
maxArea = 0
maxIndex = 0
for i, c in enumerate(contours):
area = cv.contourArea(c)
if area > maxArea:
maxArea = area
maxIndex = i
# 绘制
cv.drawContours(frame, contours, maxIndex, (255, 255, 0), 2)
# 获取外切矩形
x, y, w, h = cv.boundingRect(contours[maxIndex])
cv.rectangle(frame, (x, y), (x + w, y + h), (255, 0, 0), 2)
# 获取中心像素点
center_x = int(x + w / 2)
center_y = int(y + h / 2)
cv.circle(frame, (center_x, center_y), 5, (0, 0, 255), -1)
# 简单的打印反馈数据,之后补充运动控制
twist = Twist()
if center_x < screen_center - offset:
twist.linear.x = 0.1
twist.angular.z = 0.5
print "turn left"
elif screen_center - offset <= center_x <= screen_center + offset:
twist.linear.x = 0.3
twist.angular.z = 0
print "keep"
elif center_x > screen_center + offset:
twist.linear.x = 0.1
twist.angular.z = -0.5
print "turn right"
else:
twist.linear.x = 0
twist.angular.z = 0
print "stop"
# 将速度发出
cmd_vel_Publisher.publish(twist)
# cv.imshow("mask4", mask3)
# cv.imshow("frame", frame)
cv.waitKey(1)
rate.sleep()
ok, frame = capture.read()
来源:https://blog.csdn.net/Arcann/article/details/109531059
0
投稿
猜你喜欢
- 问:把数据从MySQL迁移到Oracle需要注意些什么?答:以下是MySQL迁到Oracle需要掌握的注意事项,希望对你有所帮助。1.自动增
- opencv > pilimport cv2 from PIL import Imageimg = cv2.imread("
- 字符串转list数组str = '1,2,3'arr = str.split(',')gpu_ids分配na
- 1.首先,我们需要过滤所有客户端提交的内容,其中包括?id=N一类,另外还有提交的html代码中的操作数据库的select及asp文件操作语
- 用Python实现点对点的聊天,2个程序,一个是client.py,一个是server.py,通过本机地址127.0.0.1连接进行通信,利
- 测试数据 http://grouplens.org/datasets/movielens/协同过滤推荐算法主要分为:1、基于用户。根据相邻用
- 有时候希望可以用js另开新窗口,但用window.open方法打开窗口总是被浏览器阻止,可以用下面的方法打开新窗口而不会遭到拦截1.新添加一
- 当然是可以的,而且非常简单,今天就教大家在ASP中不用模板生成HTML静态页的方法。这里假设有一个htmer.asp动态页面,你想把它生成为
- 代数运算定义加法运算图像叠加需要满足该条件:两幅图像的像素必须相同(尺寸,形状)减法运算乘法运算1、图像的局部显示。2、用二值蒙版图像与原图
- 本文实例讲述了Sanic框架安装与简单用法。分享给大家供大家参考,具体如下:Sanic是一个类似Flask的Python 3.5+ Web服
- 有没有想过用尺子来直接量网页上的区块间距,文字行高?屏幕标尺就是干这个的。这个功能非常适合F2E在调试样式尺寸的时候使用。打开屏幕标尺,屏幕
- 官方函数DataFrame.locAccess a group of rows and columns by label(s) or a b
- 今天写的代码片段:X = Y = []..X.append(x)Y.append(y)其中x和y是读取的每一个数据的xy值,打算将其归入列表
- Python zfill()方法返回指定长度的字符串,原字符串右对齐,前面填充0。zfill()方法语法:str.zfill(width)参
- Cookie是一种发送到客户浏览器的文本串句柄,并保存在客户机硬盘上,可以用来在某个Web站点会话之间持久地保持数据。Request和Res
- 微信小程序-拍照或选择图片并上传文件调用拍照API:https://mp.weixin.qq.com/debug/wxadoc/dev/ap
- 一、简介是一个 python 内置包,不需要额外安装即可使用urllib 是 Python 标准库中用于网络请求的库,内置四个模块,分别是u
- 1.前提你要已经安装了 某个 版本的 python, (下载地址 https://www.python.org/downloads/)安装后
- 1、生成数据集(双月数据集)class moon_data_class(object): def __init__
- PHP mysqli_rollback() 函数关闭自动提交,做一些查询,提交查询,然后回滚当前事务:<?php// 假定数据库用户名