def __init__(self,overwrite=False):
self.bridge = CvBridge()
self.overwrite = overwrite
self.data_image_topic = "/camera/data_image"
self.fd = FindData(overwrite=overwrite)
quality_low = "-preset veryfast -crf 28"
quality_med = "-preset medium -crf 23"
quality_high = "-preset veryslow -crf 16"
self.quality = quality_high
def __init__(self,overwrite=False):
self.bridge = CvBridge()
self.overwrite = overwrite
self.data_image_topic = "/camera/data_image"
self.fd = FindData(overwrite=overwrite)
self.figure = None
self.background = None
if DISPLAY_IMAGES:
cv2.namedWindow("Image FG")
cv2.namedWindow("Image Thresh")
cv2.namedWindow("Image Morphed")
def process(path_list, overwrite, tracking, video, figure):
"""
Process data
"""
fd = FindData(overwrite)
path = path_list[0]
if not tracking and not video and not figure:
tracking = True
video = True
figure = True
if figure and not tracking:
tracking = True
if tracking:
contains_data = fd.path_contains_tracking_data(path)
if not contains_data and overwrite:
print("Path does not exist or does not contain tracking data.")
elif not contains_data and not overwrite:
print(
"Path does not exist, does not contain tracking data, or tracking data has already been processed."
)
print("Try -o overwrite switch to reprocess data.")
tdp = TrackingDataProcessor(overwrite)
tdp.find_and_process_data(path)
if video:
contains_data = fd.path_contains_video_data(path)
if not contains_data and overwrite:
print("Path does not exist or does not contain video data.")
elif not contains_data and not overwrite:
print(
"Path does not exist, does not contain video data, or video data has already been processed."
)
print("Try -o overwrite switch to reprocess data.")
vdp = VideoDataProcessor(overwrite)
vdp.find_and_process_data(path)
if figure:
contains_data = fd.path_contains_figure_data(path)
if not contains_data and overwrite:
print("Path does not exist or does not contain figure data.")
elif not contains_data and not overwrite:
print(
"Path does not exist, does not contain figure data, or figure data has already been processed."
)
print("Try -o overwrite switch to reprocess data.")
fdp = FigureDataProcessor(overwrite)
fdp.find_and_process_data(path)
def __init__(self,overwrite=False):
self.overwrite = overwrite
self.fd = FindData(overwrite=overwrite)
