hofee/nbv_sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update colors

Browse Source
This commit is contained in:
Michel Breyer 2021-09-08 19:50:13 +02:00
parent d955931590
commit 83b2c173cd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
active_grasp/visualization.py
View File

@ -9,7 +9,7 @@ from vgn.utils import *
cmap = matplotlib.colors.LinearSegmentedColormap.from_list("RedGreen", ["r", "g"]) cmap = matplotlib.colors.LinearSegmentedColormap.from_list("RedGreen", ["r", "g"])
red = np.r_[1.0, 0.0, 0.0] red = np.r_[1.0, 0.0, 0.0]
blue = np.r_[31, 119, 180] / 255.0 blue = np.r_[0, 0.6, 1.0]
class Visualizer: class Visualizer:
@ -75,7 +75,7 @@ class Visualizer:
color = [color[0], color[1], color[2], alpha] color = [color[0], color[1], color[2], alpha]
self.draw(create_grasp_markers(frame, grasp, color, "best_grasp", radius=0.006)) self.draw(create_grasp_markers(frame, grasp, color, "best_grasp", radius=0.006))
def grasps(self, frame, grasps, scores, smin=0.9, smax=1.0, alpha=0.6): def grasps(self, frame, grasps, scores, smin=0.9, smax=1.0, alpha=0.8):
if len(grasps) == 0: if len(grasps) == 0:
return return