Optimize decode_outputs for OpenVINO (Megvii-BaseDetection#1535)

Avoid ScatterND ops that will cause error in openvino model optimizer.

Author: Sped0n0ld

yolox/models/yolo_head.py

@@ -246,8 +246,7 @@ def decode_outputs(self, outputs, dtype):
         grids = torch.cat(grids, dim=1).type(dtype)
         strides = torch.cat(strides, dim=1).type(dtype)
 
-        outputs[..., :2] = (outputs[..., :2] + grids) * strides
-        outputs[..., 2:4] = torch.exp(outputs[..., 2:4]) * strides
+        outputs = torch.cat([(outputs[..., 0:2] + grids) * strides, torch.exp(outputs[..., 2:4]) * strides, outputs[..., 4:]], dim=-1)
         return outputs
 
     def get_losses(