Add lidar loading utility, fixup main in motorserver.
This commit is contained in:
18
tracking/lidar_loader.py
Normal file
18
tracking/lidar_loader.py
Normal file
@@ -0,0 +1,18 @@
|
||||
from rplidar import RPLidar
|
||||
import pickle
|
||||
|
||||
|
||||
def get_scans(num_scans, device='/dev/ttyUSB0', measurements_per_scan=100):
|
||||
lidar = RPLidar(device)
|
||||
scans = lidar.iter_scans(measurements_per_scan)
|
||||
return [next(scans) for i in range(0, num_scans)]
|
||||
|
||||
|
||||
def save_scans_bytes(scans, filename='out.pickle'):
|
||||
with open(filename, 'wb') as f:
|
||||
pickle.dump(scans, f)
|
||||
|
||||
|
||||
def load_scans_bytes_file(filename):
|
||||
with open(filename, 'rb') as f:
|
||||
return pickle.load(f)
|
||||
Reference in New Issue
Block a user