def test_tracking(self):
tracker = MOTReadingAsTracker()
tracker.load_frames(root_images=_root_images, frame_num_start=1)
hooks = [
Recorder(),
MOTReader(path_mot_txt=_input_mot_txt),
ImageWriter(root_output_images=_root_output_images)
]
tracker.register_hooks(hooks)
tracker.track(0, 509)
def test_road_object_tracking_transdiff(self):
tracker = IoUTracker()
tracker.load_frames(root_images=_root_road_object_images,
frame_num_start=-1,
frame_num_end=-1)
hooks = [
RoadObjectDetection(score_thre=0.17),
BoxCoordinateNormalizer(),
ApproachingInitializer(),
FeatureMatcher(interval=6, box_remove_margin=0.1),
# TransformedMidpointCalculator(),
TransformedMidpointDifferencer(),
# HorizontalMovementCounter(right_trend_is_approaching=False),
# AreaCalculator(),
# MidpointCalculator(),
# WidthAndHeihtCalculator(),
RiskyJudger(),
Recorder(),
ImageWriter(root_output_images=_root_road_object_output_images),
ImageWriterForApproaching(
root_output_images=_root_road_object_output_approaching_images
),
VideoWriterForTracking(
root_output_video=_root_road_object_output_video,
do_resize=True,
fps=30.0),
VideoWriterForApproaching(
root_output_video=_root_road_object_output_approaching_video,
do_resize=True,
fps=30.0),
VideoWriterForMatching(
root_output_video=
_root_road_object_output_feature_matching_video,
size=(1024, 512),
do_resize=True,
fps=30.0),
MOTWriter(root_output_mot=_root_road_object_output_mot,
labels=["vehicle"]),
MOTWriterForApproaching(
root_output_mot=_root_road_object_output_approaching_mot),
MOTWriterForRisky(
root_output_mot=_root_road_object_output_risky_mot),
FeatureStatWriter(root_output_feature_stat=_root_feature_stat),
]
tracker.register_hooks(hooks)
tracker.track()
def test_mirror_tracking(self):
tracker = IoUTracker()
tracker.load_frames(root_images=_root_mirror_images,
frame_num_start=-1,
frame_num_end=-1)
hooks = [
MirrorDetection(0.3, 0.3),
Recorder(),
ImageWriter(root_output_images=_root_mirror_output_images),
VideoWriterForTracking(root_output_video=_root_mirror_output_video,
do_resize=True,
fps=30.0,
size=(3840, 2160)),
MOTWriter(root_output_mot=_root_mirror_output_mot)
]
tracker.register_hooks(hooks)
tracker.track()
def test_null_road_object_tracking(self):
tracker = NullTracker()
tracker.load_frames(root_images=_root_road_object_images,
frame_num_start=-1,
frame_num_end=-1)
hooks = [
MOTReader(path_mot_txt=_path_input_mot_txt),
BoxCoordinateNormalizer(),
ApproachingInitializer(),
FeatureMatcher(interval=30),
TransformedMidpointCalculator(),
# HorizontalMovementCounter(right_trend_is_approaching=True),
# AreaCalculator(),
# MidpointCalculator(),
# WidthAndHeihtCalculator(),
RiskyJudger(),
Recorder(),
ImageWriter(root_output_images=_root_road_object_output_images),
ImageWriterForApproaching(
root_output_images=_root_road_object_output_approaching_images
),
VideoWriterForTracking(
root_output_video=_root_road_object_output_video,
do_resize=True,
fps=30.0),
VideoWriterForApproaching(
root_output_video=_root_road_object_output_approaching_video,
do_resize=True,
fps=30.0),
VideoWriterForMatching(
root_output_video=
_root_road_object_output_feature_matching_video,
size=(1024, 512),
do_resize=True,
fps=30.0),
# MOTWriter(root_output_mot=_root_road_object_output_mot)
MOTWriterForApproaching(
root_output_mot=_root_road_object_output_approaching_mot),
MOTWriterForRisky(
root_output_mot=_root_road_object_output_risky_mot)
]
tracker.register_hooks(hooks)
tracker.track()
root_mirror_seqs=os.path.join(path_exp, seq_name, 'mirror_seqs'))
for root_name, root in roots_for_input.items():
assert os.path.isdir(root), f'Not exists: {root} for {seq_name}'
for root_name, root in roots_for_output.items():
os.makedirs(root, exist_ok=True)
tracker = IoUTracker()
tracker.load_frames(root_images=roots_for_input['root_mirror_images'],
frame_num_start=-1,
frame_num_end=-1)
hooks = [
MirrorDetection(0.3, 0.3),
Recorder(),
ImageWriter(root_output_images=roots_for_output[
'root_mirror_output_images']),
VideoWriterForTracking(
root_output_video=roots_for_output['root_mirror_output_video'],
do_resize=True,
fps=30.0,
size=(3840, 2160)),
MOTWriter(
root_output_mot=roots_for_output['root_mirror_output_mot'])
]
tracker.register_hooks(hooks)
if _IF_POSSIBLE_SKIP_MIR_RECOGNITION and (len(tracker.loader) == len(
os.listdir(roots_for_output['root_mirror_output_images']))):
logging.info('Mirror recoginition stage was skipped.')
else:
tracker.track()
frame_num_start=-1,
frame_num_end=-1)
hooks = [
RoadObjectDetection(score_thre=0.17),
BoxCoordinateNormalizer(),
ApproachingInitializer(),
FeatureMatcher(interval=6, box_remove_margin=0.1),
# TransformedMidpointCalculator(),
TransformedMidpointDifferencer(),
# HorizontalMovementCounter(right_trend_is_approaching=False),
# AreaCalculator(),
# MidpointCalculator(),
# WidthAndHeihtCalculator(),
RiskyJudger(),
Recorder(),
ImageWriter(root_output_images=roots_for_output[
'root_road_object_output_images']),
ImageWriterForApproaching(root_output_images=roots_for_output[
'root_road_object_output_approaching_images']),
VideoWriterForTracking(root_output_video=roots_for_output[
'root_road_object_output_video'],
do_resize=True,
fps=30.0),
VideoWriterForApproaching(root_output_video=roots_for_output[
'root_road_object_output_approaching_video'],
do_resize=True,
fps=30.0),
VideoWriterForMatching(root_output_video=roots_for_output[
'root_road_object_output_feature_matching_video'],
size=(1024, 512),
do_resize=True,
fps=30.0),
_root_road_object_output_risky_mot = "/home/appuser/src/pipedet/simple_demonstration/output_risky_including_road_object_recognition/mot_for_risky"
_root_feature_stat = "/home/appuser/src/pipedet/simple_demonstration/output_risky_including_road_object_recognition/feature_stat"
if __name__ == '__main__':
tracker = IoUTracker()
tracker.load_frames(root_images=_root_cropped_images, frame_num_start=-1, frame_num_end=-1)
hooks = [
RoadObjectDetection(score_thre = 0.17),
BoxCoordinateNormalizer(),
ApproachingInitializer(),
FeatureMatcher(interval=6, box_remove_margin=0.1),
# TransformedMidpointCalculator(),
TransformedMidpointDifferencer(),
# HorizontalMovementCounter(right_trend_is_approaching=False),
# AreaCalculator(),
# MidpointCalculator(),
# WidthAndHeihtCalculator(),
RiskyJudger(),
Recorder(),
ImageWriter(_root_road_object_images=_root_road_object_output_tracking_images),
ImageWriterForApproaching(root_output_images=_root_road_object_output_approaching_images),
VideoWriterForTracking(root_output_video=_root_road_object_output_video_for_tracking, do_resize=True, fps=30.0),
VideoWriterForApproaching(root_output_video=_root_road_object_output_approaching_video, do_resize=True, fps=30.0),
VideoWriterForMatching(root_output_video=_root_road_object_output_feature_matching_video, size=(1024,512) ,do_resize=True, fps=30.0),
MOTWriter(root_output_mot=_root_road_object_output_tracking_mot, labels=["vehicle"]),
MOTWriterForApproaching(root_output_mot=_root_road_object_output_approaching_mot),
MOTWriterForRisky(root_output_mot=_root_road_object_output_risky_mot),
FeatureStatWriter(root_output_feature_stat=_root_feature_stat),
]
tracker.register_hooks(hooks)
tracker.track()
from pipedet.solver.get_mirror_seq import MirrorSeq
"""
Mirror tracking demonstration.
"""
_root_in_vehicle_frames = '/home/appuser/src/pipedet/simple_demonstration/demo_in_vehicle_frame'
_root_output_video_for_tracking = '/home/appuser/src/pipedet/simple_demonstration/output_mirror_tracking/video_for_tracking'
_root_output_images_for_tracking = '/home/appuser/src/pipedet/simple_demonstration/output_mirror_tracking/images_for_tracking'
_root_output_mot_for_tracking = '/home/appuser/src/pipedet/simple_demonstration/output_mirror_tracking/mot_for_tracking'
if __name__ == '__main__':
tracker = IoUTracker()
tracker.load_frames(root_images=_root_in_vehicle_frames,
frame_num_start=-1,
frame_num_end=-1)
hooks = [
MirrorDetection(0.3, 0.3),
Recorder(),
ImageWriter(root_output_images=_root_output_images_for_tracking),
VideoWriterForTracking(
root_output_video=_root_output_video_for_tracking,
do_resize=True,
fps=30.0,
size=(3840, 2160)),
MOTWriter(root_output_mot=_root_output_mot_for_tracking)
]
tracker.register_hooks(hooks)
tracker.track()