thai thong commited on
Commit
e7b0d30
·
1 Parent(s): c412fb1

change gpu to cpu

Browse files
Files changed (1) hide show
  1. detect_deepsort.py +1 -1
detect_deepsort.py CHANGED
@@ -40,7 +40,7 @@ def initialize_deepsort():
40
  max_age=cfg_deep.DEEPSORT.MAX_AGE, n_init=cfg_deep.DEEPSORT.N_INIT,
41
  #nn_budget: It sets the budget for the nearest-neighbor search.
42
  nn_budget=cfg_deep.DEEPSORT.NN_BUDGET,
43
- use_cuda=True
44
  )
45
 
46
  return deepsort
 
40
  max_age=cfg_deep.DEEPSORT.MAX_AGE, n_init=cfg_deep.DEEPSORT.N_INIT,
41
  #nn_budget: It sets the budget for the nearest-neighbor search.
42
  nn_budget=cfg_deep.DEEPSORT.NN_BUDGET,
43
+ use_cuda=False
44
  )
45
 
46
  return deepsort