oskarastrom commited on
Commit
9c86395
1 Parent(s): d8d9ab6

Update tracker.py

Browse files
Files changed (1) hide show
  1. lib/fish_eye/tracker.py +4 -0
lib/fish_eye/tracker.py CHANGED
@@ -30,12 +30,15 @@ class Tracker:
30
  conf = 0
31
  min_score = 1000000
32
  if self.algorithm == Sort:
 
33
  for det in dets:
34
  score = sum(abs(det[0:4] - track[0:4]))
35
  if (score < min_score):
36
  min_score = score
37
  conf = det[4]
 
38
  elif self.algorithm == Associate:
 
39
  for det in dets[0]:
40
  score = sum(abs(det[0:4] - track[0:4]))
41
  if (score < min_score):
@@ -46,6 +49,7 @@ class Tracker:
46
  if (score < min_score):
47
  min_score = score
48
  conf = det[4]
 
49
 
50
  # Assign Track
51
  self.fish_ids[int(track[4])] += 1
 
30
  conf = 0
31
  min_score = 1000000
32
  if self.algorithm == Sort:
33
+ print("SORT")
34
  for det in dets:
35
  score = sum(abs(det[0:4] - track[0:4]))
36
  if (score < min_score):
37
  min_score = score
38
  conf = det[4]
39
+ print(conf)
40
  elif self.algorithm == Associate:
41
+ print("Associate")
42
  for det in dets[0]:
43
  score = sum(abs(det[0:4] - track[0:4]))
44
  if (score < min_score):
 
49
  if (score < min_score):
50
  min_score = score
51
  conf = det[4]
52
+ print(conf)
53
 
54
  # Assign Track
55
  self.fish_ids[int(track[4])] += 1