From 97b66c14d46cec187c583824555392ebfecec94a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mat=C4=9Bj=20=C5=A0m=C3=ADd?= Date: Fri, 22 Feb 2019 09:03:06 +0100 Subject: [PATCH 1/2] fix batch generation fix failed assertion: deepmatching.py(73)get_match(): assert frame1 <= frame2, "First frame must be greater/equal" --- cabbage/features/combined.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cabbage/features/combined.py b/cabbage/features/combined.py index 3207254..d616b18 100644 --- a/cabbage/features/combined.py +++ b/cabbage/features/combined.py @@ -107,7 +107,7 @@ def gen_feature_batch(batch, lookup, dmax, dm, reid, W, video_name): aabb_j, aabb_i = aabb_j[IN_RANGE], aabb_i[IN_RANGE] scores_i, scores_j = scores_i[IN_RANGE], scores_j[IN_RANGE] - frame_i, framej = frame_i[IN_RANGE], frame_j[IN_RANGE] + frame_i, frame_j = frame_i[IN_RANGE], frame_j[IN_RANGE] i = i[IN_RANGE]; j = j[IN_RANGE] Im_j, Im_i = \ From ea29ff4eb4ff7db1a7fa5e1842162d834f9fdb75 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mat=C4=9Bj=20=C5=A0m=C3=ADd?= Date: Fri, 22 Feb 2019 09:03:20 +0100 Subject: [PATCH 2/2] fix StackNet loading --- cabbage/MultiplePeopleTracking.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cabbage/MultiplePeopleTracking.py b/cabbage/MultiplePeopleTracking.py index 52e2e6a..377f0ec 100644 --- a/cabbage/MultiplePeopleTracking.py +++ b/cabbage/MultiplePeopleTracking.py @@ -51,7 +51,7 @@ def execute_multiple_people_tracking(video_folder, X, Dt, video_name, dmax, sett dm = DeepMatching(deep_matching_binary, data_root, dmax) dm.generate_matches(video_folder, video_name) - reid = StackNet64x64(root) + reid = StackNet64x64(data_root) gg = BatchGraphGenerator(data_root, reid, dm, dmax, video_name) gg.build(Dt, X, W, batch_size=batch_size)