ruinmessi Feng Wang commited on
Commit
a4c8b2d
·
1 Parent(s): f3ca1f5

Fix eval bug (#161)

Browse files

* change to subprocess beckend

* fix eval bug

* Update launch.py

make flake8 happy

Co-authored-by: Feng Wang <[email protected]>

Files changed (1) hide show
  1. yolox/data/data_augment.py +1 -1
yolox/data/data_augment.py CHANGED
@@ -266,4 +266,4 @@ class ValTransform:
266
  # assume input is cv2 img for now
267
  def __call__(self, img, res, input_size):
268
  img, _ = preproc(img, input_size, self.means, self.std, self.swap)
269
- return torch.tensor_as(img), torch.zeros(1, 5)
 
266
  # assume input is cv2 img for now
267
  def __call__(self, img, res, input_size):
268
  img, _ = preproc(img, input_size, self.means, self.std, self.swap)
269
+ return torch.as_tensor(img), torch.zeros(1, 5)