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()
总结
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
暂无评论...
RTX 5090要首发 性能要翻倍!三星展示GDDR7显存
三星在GTC上展示了专为下一代游戏GPU设计的GDDR7内存。
首次推出的GDDR7内存模块密度为16GB,每个模块容量为2GB。其速度预设为32 Gbps(PAM3),但也可以降至28 Gbps,以提高产量和初始阶段的整体性能和成本效益。
据三星表示,GDDR7内存的能效将提高20%,同时工作电压仅为1.1V,低于标准的1.2V。通过采用更新的封装材料和优化的电路设计,使得在高速运行时的发热量降低,GDDR7的热阻比GDDR6降低了70%。
更新日志
2024年11月23日
2024年11月23日
- 凤飞飞《我们的主题曲》飞跃制作[正版原抓WAV+CUE]
- 刘嘉亮《亮情歌2》[WAV+CUE][1G]
- 红馆40·谭咏麟《歌者恋歌浓情30年演唱会》3CD[低速原抓WAV+CUE][1.8G]
- 刘纬武《睡眠宝宝竖琴童谣 吉卜力工作室 白噪音安抚》[320K/MP3][193.25MB]
- 【轻音乐】曼托凡尼乐团《精选辑》2CD.1998[FLAC+CUE整轨]
- 邝美云《心中有爱》1989年香港DMIJP版1MTO东芝首版[WAV+CUE]
- 群星《情叹-发烧女声DSD》天籁女声发烧碟[WAV+CUE]
- 刘纬武《睡眠宝宝竖琴童谣 吉卜力工作室 白噪音安抚》[FLAC/分轨][748.03MB]
- 理想混蛋《Origin Sessions》[320K/MP3][37.47MB]
- 公馆青少年《我其实一点都不酷》[320K/MP3][78.78MB]
- 群星《情叹-发烧男声DSD》最值得珍藏的完美男声[WAV+CUE]
- 群星《国韵飘香·贵妃醉酒HQCD黑胶王》2CD[WAV]
- 卫兰《DAUGHTER》【低速原抓WAV+CUE】
- 公馆青少年《我其实一点都不酷》[FLAC/分轨][398.22MB]
- ZWEI《迟暮的花 (Explicit)》[320K/MP3][57.16MB]