Python DrawPrediction示例

示例#1

    def debug_images(self, pose):

        c_img = self.cam.read_color_data()
        dp = DrawPrediction()

        image = dp.draw_prediction(np.copy(c_img), pose)

        cv2.imshow('debug', image)
        cv2.waitKey(30)

示例#2

文件： gripper.py 项目： mdlaskey/IL_ROS_HSR

    def plot_on_true(self, pose, true_img):

        #pose = self.convert_crop(pose)

        dp = DrawPrediction()

        image = dp.draw_prediction(np.copy(true_img), pose)

        cv2.imshow('label_given', image)

        cv2.waitKey(30)

示例#3

from il_ros_hsr.p_pi.bed_making.gripper import Bed_Gripper
from il_ros_hsr.p_pi.bed_making.table_top import TableTop
from il_ros_hsr.core.web_labeler import Web_Labeler
from il_ros_hsr.core.python_labeler import Python_Labeler

from il_ros_hsr.p_pi.bed_making.check_success import Success_Check
from il_ros_hsr.p_pi.bed_making.self_supervised import Self_Supervised
import il_ros_hsr.p_pi.bed_making.config_bed as cfg
import cPickle as pickle
import os

from il_ros_hsr.core.rgbd_to_map import RGBD2Map

from data_aug.draw_cross_hair import DrawPrediction

dp = DrawPrediction()
#latest, 46-49 from rollout_dart
sm = 0

wl = Python_Labeler()
for rnum in range(0, 10):
    # path = cfg.STAT_PATH+'stat_' + str(rnum) + '/rollout.p'
    path = cfg.STAT_PATH + 'stat_' + str(rnum) + '/rollout.p'

    #IPython.embed()
    labeled_data = []
    if os.path.exists(path):
        data = pickle.load(open(path, 'rb'))

        count = 0

示例#4

from il_ros_hsr.p_pi.bed_making.gripper import Bed_Gripper
from il_ros_hsr.p_pi.bed_making.table_top import TableTop
from il_ros_hsr.core.web_labeler import Web_Labeler
from il_ros_hsr.core.python_labeler import Python_Labeler

from il_ros_hsr.p_pi.bed_making.check_success import Success_Check
from il_ros_hsr.p_pi.bed_making.self_supervised import Self_Supervised
import il_ros_hsr.p_pi.bed_making.config_bed as cfg
import cPickle as pickle

from il_ros_hsr.core.rgbd_to_map import RGBD2Map

from data_aug.draw_cross_hair import DrawPrediction

dp = DrawPrediction()

for rnum in range(0, 1):
    path = cfg.STAT_PATH + 'stat_' + str(rnum) + '/rollout.p'
    data = pickle.load(open(path, 'rb'))
    print(data)
    count = 0
    for datum in data:

        if type(datum) == list:
            continue

        if datum['type'] == 'grasp':
            pose = datum['net_pose']
            c_img = datum['c_img']

示例#5

文件： analytic_grasp.py 项目： mdlaskey/IL_ROS_HSR

 def __init__(self):
     self.supp = Analytic_Supp()
     self.dp = DrawPrediction()

示例#6

文件： movie_data.py 项目： mdlaskey/IL_ROS_HSR

from il_ros_hsr.p_pi.bed_making.gripper import Bed_Gripper
from il_ros_hsr.p_pi.bed_making.table_top import TableTop
from il_ros_hsr.core.web_labeler import Web_Labeler
from il_ros_hsr.core.python_labeler import Python_Labeler

from il_ros_hsr.p_pi.bed_making.check_success import Success_Check
from il_ros_hsr.p_pi.bed_making.self_supervised import Self_Supervised
import il_ros_hsr.p_pi.bed_making.config_bed as cfg
import cPickle as pickle

from il_ros_hsr.core.rgbd_to_map import RGBD2Map

from data_aug.draw_cross_hair import DrawPrediction

dp = DrawPrediction()
#latest, 46-49 from rollout_dart

MOVIE_PATH = 'paper_dart_data/'

for rnum in range(10, 30):
    # path = cfg.STAT_PATH+'stat_' + str(rnum) + '/rollout.p'
    path = cfg.ROLLOUT_PATH + 'rollout_' + str(rnum) + '/rollout.p'

    data = pickle.load(open(path, 'rb'))
    print(data)
    count = 0
    for datum in data:

        if type(datum) == list:
            continue