kenaug74/mycobot_ros
1
0
Fork 0
mirror of https://github.com/elephantrobotics/mycobot_ros.git synced 2026-07-14 09:00:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add red,gray,blue,green folder

Browse source
This commit is contained in:
wangWking 2022-05-05 18:37:50 +08:00
parent cca5330483
commit b631c7b222
1 changed files with 41 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
mycobot_ai/scripts/add_img.py
View file

@ -1,4 +1,6 @@
# coding:utf-8 # coding:utf-8
from ast import keyword
from fileinput import filename
import os, cv2, sys import os, cv2, sys
@ -11,8 +13,8 @@ def take_photo():
print("*********************************************") print("*********************************************")
# 创建/使用local_photo文件夹 # 创建/使用local_photo文件夹
class_name = "local_photo" class_name = "res"
if (os.path.exists("local_photo")): if (os.path.exists("res")):
pass pass
else: else:
os.mkdir(class_name) os.mkdir(class_name)
@ -47,9 +49,22 @@ def take_photo():
def cut_photo(): def cut_photo():
path = os.getcwd() + '/local_photo/img'
for i, j, k in os.walk(path): path_red = '/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/red'
file_len = len(k) for i, j, k in os.walk(path_red):
file_len_red = len(k)
path_gray = '/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/gray'
for i, j, k in os.walk(path_gray):
file_len_gray = len(k)
path_green = '/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/green'
for i, j, k in os.walk(path_green):
file_len_green = len(k)
path_blue = '/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/blue'
for i, j, k in os.walk(path_blue):
file_len_blue = len(k)
print("请截取要识别的部分") print("请截取要识别的部分")
# root = tk.Tk() # root = tk.Tk()
# root.withdraw() # root.withdraw()
@ -58,7 +73,7 @@ def cut_photo():
# temp2= cv.cvtColor(np.asarray(temp2),cv.COLOR_RGB2BGR) # temp2= cv.cvtColor(np.asarray(temp2),cv.COLOR_RGB2BGR)
# cut = np.array(temp2) # cut = np.array(temp2)
cut = cv2.imread(r"local_photo/takephoto.jpeg") cut = cv2.imread(r"res/takephoto.jpeg")
cv2.imshow('original', cut) cv2.imshow('original', cut)
# C:\Users\Elephant\Desktop\pymycobot+opencv\local_photo/takephoto.jpeg # C:\Users\Elephant\Desktop\pymycobot+opencv\local_photo/takephoto.jpeg
@ -71,13 +86,30 @@ def cut_photo():
x, y, w, h = roi x, y, w, h = roi
print(roi) print(roi)
kw = input("请输入保存图片文件夹名称('red','gray','blue','green'）:")
# print(kw)
# 显示ROI并保存图片 # 显示ROI并保存图片
if roi != (0, 0, 0, 0): if roi != (0, 0, 0, 0):
crop = cut[y:y + h, x:x + w] crop = cut[y:y + h, x:x + w]
cv2.imshow('crop', crop) cv2.imshow('crop', crop)
cv2.imwrite('local_photo/img/goal{}.jpeg'.format(str(file_len + 1)), # 选择红桶文件夹
crop) if kw == 'red':
print('Saved!') cv2.imwrite('/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/red/goal{}.jpeg'.format(str(file_len_red + 1)),crop)
print('Saved')
# 选择灰桶文件夹
elif kw == 'gray':
cv2.imwrite('/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/gray/goal{}.jpeg'.format(str(file_len_gray+1)),crop)
print('Saved')
# 选择绿桶文件夹
elif kw == 'green':
cv2.imwrite('/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/green/goal{}.jpeg'.format(str(file_len_green+1)),crop)
print('Saved')
# 选择蓝桶文件夹
elif kw == 'blue':
cv2.imwrite('/home/h/catkin_ws/src/mycobot_ros/mycobot_ai/res/blue/goal{}.jpeg'.format(str(file_len_blue+1)),crop)
print('Saved')
# 退出 # 退出
cv2.waitKey(0) cv2.waitKey(0)