-
Notifications
You must be signed in to change notification settings - Fork 3
/
error_plotter.py
50 lines (43 loc) · 1.3 KB
/
error_plotter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import numpy as np
from matplotlib import pyplot as plt
import pickle
"""Used for plotting successful and unsuccesful insertions"""
file = './errors_testing'
name = file + '.pkl'
with open(name, 'rb') as f:
loaded_dict = pickle.load(f)
print()
# success = np.array(loaded_dict.get('success'))
errors = np.array(loaded_dict.get('error'))
peg_radius = 2.1/1000
hole_radius = 2.4/1000
r_low = 3.3/1000
r_high = 3.6/1000
R_hole = hole_radius
R_peg = peg_radius
theta = np.linspace(0, 2*np.pi, 1000)
x_hole = R_hole*np.cos(theta)
y_hole = R_hole*np.sin(theta)
x_peg = R_peg*np.cos(theta)
y_peg = R_peg
x_low = r_low*np.cos(theta)
y_low = r_low*np.sin(theta)
x_high = r_high*np.cos(theta)
y_high = r_high*np.sin(theta)
plt.figure()
# plt.title(file + ': 100 eval ep: ring error: [0.6, 0.8]')
plt.plot(x_hole, y_hole, color='k', label='hole', linewidth=3)
plt.plot(x_low, y_low, color='r', linestyle='dashed')
plt.plot(x_high, y_high, color='r', linestyle='--')
plt.axhline(y=0, color='k', linestyle='dotted')
plt.axvline(x=0, color='k', linestyle='dotted')
for i in range(len(errors)):
err = errors[i]
# suc = success[i]
# if suc:
# plt.scatter(err[0], err[1], c='green')
# else:
plt.scatter(err[0], err[1], c='red')
# print(f'Unsuccessful insertion for: {err[:2]*1000}mm')
plt.grid()
plt.show()