tiny adaptations of the package

This commit is contained in:
Jan Grewe 2021-06-10 09:45:17 +02:00
parent c3b37dfe7d
commit 16873702d4
3 changed files with 7 additions and 2 deletions

View File

@ -1,2 +1,2 @@
from .image_marker import ImageMarker, MarkerTask
from .tracking_result import TrackingResult
from .tracking_result import TrackingResult, coordinate_transformation

View File

@ -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:

View File

@ -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):