Merge pull request #911 from mikel-brostrom/fix-appearance-based-meth…

…ods-on-cpu

add @torch.no_grad() to all ReID related inference methods

boxmot/botsort/bot_sort.py

@@ -7,6 +7,7 @@
 from .gmc import GMC
 from .basetrack import BaseTrack, TrackState
 from .kalman_filter import KalmanFilter
+import torch
 
 # from fast_reid.fast_reid_interfece import FastReIDInterface
 
@@ -482,6 +483,7 @@ def _xywh_to_xyxy(self, bbox_xywh):
         y2 = min(int(y + h / 2), self.height - 1)
         return x1, y1, x2, y2
 
+    @torch.no_grad()
     def _get_features(self, bbox_xywh, ori_img):
         im_crops = []
         for box in bbox_xywh:

boxmot/deepocsort/ocsort.py

@@ -3,7 +3,7 @@
 """
 from __future__ import print_function
 
-
+import torch
 import numpy as np
 from .association import *
 from .cmc import CMCComputer
@@ -528,6 +528,7 @@ def _xywh_to_xyxy(self, bbox_xywh):
         y2 = min(int(y + h / 2), self.height - 1)
         return x1, y1, x2, y2
 
+    @torch.no_grad()
     def _get_features(self, bbox_xyxy, ori_img):
         im_crops = []
         for box in bbox_xyxy:

boxmot/strongsort/strong_sort.py

@@ -128,6 +128,7 @@ def _xyxy_to_tlwh(self, bbox_xyxy):
         h = int(y2 - y1)
         return t, l, w, h
 
+    @torch.no_grad()
     def _get_features(self, bbox_xywh, ori_img):
         im_crops = []
         for box in bbox_xywh: