oskarastrom commited on
Commit
c72ec10
·
1 Parent(s): 1446d23

Update tracker.py

Browse files
Files changed (1) hide show
  1. lib/fish_eye/tracker.py +2 -2
lib/fish_eye/tracker.py CHANGED
@@ -30,7 +30,7 @@ class Tracker:
30
  conf = 0
31
  min_score = 1000000
32
  print("algorithm", self.algorithm)
33
- if self.algorithm == Sort:
34
  print("SORT")
35
  for det in dets:
36
  score = sum(abs(det[0:4] - track[0:4]))
@@ -38,7 +38,7 @@ class Tracker:
38
  min_score = score
39
  conf = det[4]
40
  print(conf)
41
- elif self.algorithm == Associate:
42
  print("Associate")
43
  for det in dets[0]:
44
  score = sum(abs(det[0:4] - track[0:4]))
 
30
  conf = 0
31
  min_score = 1000000
32
  print("algorithm", self.algorithm)
33
+ if type(self.algorithm) == Sort:
34
  print("SORT")
35
  for det in dets:
36
  score = sum(abs(det[0:4] - track[0:4]))
 
38
  min_score = score
39
  conf = det[4]
40
  print(conf)
41
+ elif type(self.algorithm) == Associate:
42
  print("Associate")
43
  for det in dets[0]:
44
  score = sum(abs(det[0:4] - track[0:4]))