ntsc207 commited on
Commit
bae2e9d
·
verified ·
1 Parent(s): 8f4aee7

Update detect_strongsort.py

Browse files
Files changed (1) hide show
  1. detect_strongsort.py +1 -2
detect_strongsort.py CHANGED
@@ -137,13 +137,12 @@ def run_strongsort(
137
  cfg.merge_from_file(config_strongsort)
138
 
139
  # Create as many strong sort instances as there are video sources
140
- gpu = '0'
141
  strongsort_list = []
142
  for i in range(bs):
143
  strongsort_list.append(
144
  StrongSORT(
145
  strong_sort_weights,
146
- gpu,
147
  half,
148
  max_dist=cfg.STRONGSORT.MAX_DIST,
149
  max_iou_distance=cfg.STRONGSORT.MAX_IOU_DISTANCE,
 
137
  cfg.merge_from_file(config_strongsort)
138
 
139
  # Create as many strong sort instances as there are video sources
 
140
  strongsort_list = []
141
  for i in range(bs):
142
  strongsort_list.append(
143
  StrongSORT(
144
  strong_sort_weights,
145
+ device,
146
  half,
147
  max_dist=cfg.STRONGSORT.MAX_DIST,
148
  max_iou_distance=cfg.STRONGSORT.MAX_IOU_DISTANCE,