In [ ]:
!baxter.sh
!roslaunch kinect2_bridge kinect2_bridge.launch
In [ ]:
from utils_io import read_json, load_items
params = read_json('parameters.json')
ITEM_FOLDER = params['item_folder']
items = load_items(ITEM_FOLDER)
In [ ]:
import rospy
from sensor_msgs.msg import Image
from cv_bridge import CvBridge, CvBridgeError
bridge = CvBridge()
In [ ]:
def callback(msg):
global color_img
global timestamp
timestamp = msg.header.stamp
color_img = bridge.imgmsg_to_cv2(msg, "rgb8")
In [ ]:
rospy.init_node('camera_subscriber', anonymous=True)
rospy.Subscriber("/kinect2/hd/image_color_rect", Image, callback)
In [ ]:
import matplotlib.pyplot as plt
%matplotlib inline
In [ ]:
bin_img = color_img[250:700,1000:,:]
plt.imshow(bin_img);
In [ ]:
from utils_sift import compute_sift
(kp, des) = compute_sift(bin_img)
In [ ]:
n = 1
In [ ]:
from utils_sift import compute_sift, match_items
bin_img = color_img[250:700,1000:,:]
(kp, des) = compute_sift(bin_img, debug=False)
item_d, recognised_items, mask_items = match_items(bin_img, kp, des, items[n:n+1])
In [ ]:
n += 1
items[n]
In [ ]:
In [ ]:
In [ ]:
In [ ]:
In [ ]:
In [ ]:
In [ ]:
In [ ]: