ntsc207 commited on
Commit
fee9fde
·
verified ·
1 Parent(s): d980547

Update detect_deepsort.py

Browse files
Files changed (1) hide show
  1. detect_deepsort.py +1 -1
detect_deepsort.py CHANGED
@@ -164,7 +164,7 @@ def run_deepsort(
164
  max_age=cfg_deep.DEEPSORT.MAX_AGE, n_init=cfg_deep.DEEPSORT.N_INIT,
165
  #nn_budget: It sets the budget for the nearest-neighbor search.
166
  nn_budget=cfg_deep.DEEPSORT.NN_BUDGET,
167
- use_cuda=False
168
  )
169
 
170
  # Load model
 
164
  max_age=cfg_deep.DEEPSORT.MAX_AGE, n_init=cfg_deep.DEEPSORT.N_INIT,
165
  #nn_budget: It sets the budget for the nearest-neighbor search.
166
  nn_budget=cfg_deep.DEEPSORT.NN_BUDGET,
167
+ use_cuda=True
168
  )
169
 
170
  # Load model