def CreateDatasetFromRosbag(rosbagName,
pickleName,
isBebop=True,
start_frame=None,
end_frame=None):
"""Converts rosbag to format suitable for training/testing.
if start_frame, end_frame are unknown, FrameSelector will help you choose how to trim the video
Parameters
----------
rosbagName : str
The file location of the rosbag
pickleName : str
name of the new .pickle file
isBebop : bool, optional
True if you want RGB dataset for the bebop, False if you want Himax-tailored dataset
start_frame : int, optional
if known, the timestamp in ns of the frame you wish to start from
end_frame : int, optional
if known, the timestamp in ns of the frame you wish to finish at
"""
dc = DatasetCreator(rosbagName)
if (start_frame is None) or (end_frame is None):
start_frame, end_frame = dc.FrameSelector()
if isBebop == True:
dc.CreateBebopDataset(0, pickleName, start_frame, end_frame)
else:
dc.CreateHimaxDataset(config.himax_delay, pickleName, start_frame,
end_frame)
def TestDatasetCreator():
subject_name = "davide1"
dc = DatasetCreator(config.folder_path + "/data/Hand/" + subject_name +
".bag")
start_frame, end_frame = dc.FrameSelector()
dc.CreateBebopDataset(
0, config.folder_path + "/data/Hand/" + subject_name + "Hand.pickle",
start_frame, end_frame)
subject_name = "davide2"
dc2 = DatasetCreator(config.folder_path + "/data/Hand/" + subject_name +
".bag")
start_frame, end_frame = dc2.FrameSelector()
dc2.CreateBebopDataset(
0, config.folder_path + "/data/Hand/" + subject_name + "Hand.pickle",
start_frame, end_frame)
folderPath = config.folder_path + "/data/Hand/"
fileList = ["davide1Hand.pickle", "davide2Hand.pickle"]
DatasetCreator.JoinPickleFiles(
fileList, config.folder_path + "/data/Hand/DavideHand.pickle",
folderPath)
