jgrewe/tracking_tools
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[test] split and fix tasks

Browse Source
This commit is contained in:
Jan Grewe 2020-12-19 18:35:16 +01:00
parent 185e348a0b
commit 011db18f46
1 changed files with 24 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
test.py
View File

@ -5,16 +5,34 @@ from tracking_result import TrackingResult
import image_marker as im
def main(filename):
tr = TrackingResult(filename)
def show_tracking_results(dlc_results_file):
tr = TrackingResult(dlc_results_file)
tr.plot(bodypart="snout")
t, x, y, l = tr.position_values(framerate=30)
print("frame_count", len(t))
print("min x pos: %.2f, max x pos: %.2f" % (np.min(x), np.max(x)))
print("min y pos: %.2f, max y pos: %.2f" % (np.min(y), np,max(y)))
print("min likelihood: %.2f, max likelihood: %.2f" % (np.min(l), np,max(l)))
def show_image_marking(video_file):
tank_task = im.MarkerTask("tank limits", ["bottom left corner", "top left corner", "top right corner", "bottom right corner"], "Mark tank corners")
feeder_task = im.MarkerTask("Feeder positions", list(map(str, range(1, 2))), "Mark feeder positions")
feeder_task = im.MarkerTask("Feeder positions", list(map(str, range(1, 9))), "Mark feeder positions", color="tab:red", marker="s")
tasks = [tank_task, feeder_task]
image_marker = im.ImageMarker(tasks)
marker_positions = image_marker.mark_movie(filename, 1)
print(marker_positions)
marker_positions = image_marker.mark_movie(video_file, 1)
for t in marker_positions:
print(t)
def main(dlc_results_file, video_file):
show_tracking_results(dlc_results_file)
show_image_marking(video_file)
if __name__ == "__main__":
filename = "2020.12.11_lepto48DLC_resnet50_boldnessDec11shuffle1_200000.h5"
main(filename)
video = "2020.12.11_lepto48DLC_resnet50_boldnessDec11shuffle1_200000_labeled.mp4"
main(filename, video)