tami/MYNT-EYE-S-SDK
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Upgrade analytics tools with mynteye dataset

Browse Source
This commit is contained in:
John Zhao
2018-04-15 21:32:14 +08:00
parent 2d107d7392
commit ad11dfe6e1
6 changed files with 168 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tools/analytics/imu_analytics.py
View File

@@ -11,7 +11,7 @@ TOOLBOX_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(os.path.join(TOOLBOX_DIR, 'internal'))
# pylint: disable=import-error,wrong-import-position
from data import DataError, Dataset, ROSBag, What
from data import DataError, Dataset, ROSBag, MYNTEYE, What
TIME_SCALE_FACTORS = {
@@ -626,9 +626,13 @@ def _main():
if args.config:
import yaml
config = yaml.load(file(args.config, 'r'))
if config['dataset'] != 'rosbag':
sys.exit('Error: dataset model only support rosbag now')
dataset = ROSBag(path, **config['rosbag'])
model = config['dataset']
if model == 'rosbag':
dataset = ROSBag(path, **config['rosbag'])
elif model == 'mynteye':
dataset = MYNTEYE(path)
else:
sys.exit('Error: dataset model not supported {}'.format(model))
else:
dataset = ROSBag(
path,

12
tools/analytics/stamp_analytics.py
View File

@@ -11,7 +11,7 @@ TOOLBOX_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(os.path.join(TOOLBOX_DIR, 'internal'))
# pylint: disable=import-error,wrong-import-position
from data import ROSBag, What
from data import ROSBag, MYNTEYE, What
ANGLE_DEGREES = 'd'
@@ -311,9 +311,13 @@ def _main():
if args.config:
import yaml
config = yaml.load(file(args.config, 'r'))
if config['dataset'] != 'rosbag':
sys.exit('Error: dataset model only support rosbag now')
dataset = ROSBag(path, **config['rosbag'])
model = config['dataset']
if model == 'rosbag':
dataset = ROSBag(path, **config['rosbag'])
elif model == 'mynteye':
dataset = MYNTEYE(path)
else:
sys.exit('Error: dataset model not supported {}'.format(model))
else:
dataset = ROSBag(path,
topic_img_left='/mynteye/left',