Commit 99b1453b authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed bug on rule_based annotation

parent f1686c59
......@@ -651,7 +651,8 @@ class coordinates:
speeds = self.get_coords(speed=1)
for key in tqdm(self._tables.keys()):
video = [vid for vid in self._videos if key in vid][0]
video = [vid for vid in self._videos if key + "DLC" in vid][0]
print(key, video)
tag_dict[key] = deepof.pose_utils.rule_based_tagging(
list(self._tables.keys()),
self._videos,
......
......@@ -136,7 +136,6 @@ def climb_wall(
if arena_type == "circular":
center = np.zeros(2) if centered_data else np.array(arena[:2])
radius = arena[2]
print(radius)
climbing = np.linalg.norm(nose - center, axis=1) > (radius + tol)
else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment