Skip to content

Commit f7d06db

Browse files
author
arch
committed
fix crashe due to empty seqence
1 parent e6cf15a commit f7d06db

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

funscript_editor/algorithms/funscriptgenerator.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -232,6 +232,13 @@ def calculate_score(self, bboxes) -> None:
232232
if self.params.track_men:
233233
men_center = [self.get_center(item) for item in bboxes['Men'][tracker_number]]
234234

235+
if len(men_center) == 0:
236+
score['x'][tracker_number] = []
237+
score['y'][tracker_number] = []
238+
score['distance'][tracker_number] = []
239+
score['roll'][tracker_number] = []
240+
continue
241+
235242
score['x'][tracker_number] = np.array([w[0] - m[0] for w, m in zip(woman_center, men_center)])
236243
score['y'][tracker_number] = np.array([m[1] - w[1] for w, m in zip(woman_center, men_center)])
237244

0 commit comments

Comments
 (0)