Taino commited on
Commit
59009ce
·
verified ·
1 Parent(s): 90fc73e

Update detection.py

Browse files
Files changed (1) hide show
  1. detection.py +1 -1
detection.py CHANGED
@@ -27,7 +27,7 @@ face_app.prepare(ctx_id=0) # Prepare for GPU inference
27
  reid_extractor = torchreid.utils.FeatureExtractor(
28
  model_name='osnet_x0_25',
29
  model_path='osnet_x0_25_market1501.pth', # Pre-trained on Market1501 dataset
30
- device='cuda' if torch.cuda.is_available() else 'cpu'
31
  )
32
 
33
  # =============================================================================
 
27
  reid_extractor = torchreid.utils.FeatureExtractor(
28
  model_name='osnet_x0_25',
29
  model_path='osnet_x0_25_market1501.pth', # Pre-trained on Market1501 dataset
30
+ device='cuda'
31
  )
32
 
33
  # =============================================================================