diff --git a/etrack/__init__.py b/etrack/__init__.py index 486c4ba..f7a4541 100644 --- a/etrack/__init__.py +++ b/etrack/__init__.py @@ -1,2 +1,2 @@ from .image_marker import ImageMarker, MarkerTask -from .tracking_result import TrackingResult \ No newline at end of file +from .tracking_result import TrackingResult, coordinate_transformation \ No newline at end of file diff --git a/etrack/image_marker.py b/etrack/image_marker.py index 8ad7bad..49bc894 100644 --- a/etrack/image_marker.py +++ b/etrack/image_marker.py @@ -17,7 +17,7 @@ class ImageMarker: self._fig.canvas.mpl_connect('button_press_event', self._on_click_event) self._fig.canvas.mpl_connect('close_event', self._fig_close_event) self._fig.canvas.mpl_connect('key_press_event', self._key_press_event) - + def mark_movie(self, filename, frame_number=0): if not os.path.exists(filename): raise IOError("file %s does not exist!" % filename) @@ -30,6 +30,7 @@ class ImageMarker: print("Reading frame: %i" % frame_counter, end="\r") success, frame = video.read() frame_counter += 1 + if success: self._fig.gca().imshow(frame) else: diff --git a/etrack/tracking_result.py b/etrack/tracking_result.py index 2e7d3e2..1d3bee9 100644 --- a/etrack/tracking_result.py +++ b/etrack/tracking_result.py @@ -14,6 +14,10 @@ y_factor = 0.81/height # Einheit m/px center = (np.round(x_0 + width/2), np.round(y_0 + height/2)) center_meter = ((center[0] - x_0) * x_factor, (center[1] - y_0) * y_factor) """ +def coordinate_transformation(position,x_0, y_0, x_factor, y_factor): + x = (position[0] - x_0) * x_factor + y = (position[1] - y_0) * y_factor + return (x, y) #in m class TrackingResult(object):