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]>
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.
|
|
|
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)
|