Update tools/demo_api.py

tools/demo_api.py

@@ -74,7 +74,7 @@ class Predictor(object):
         img_info["raw_img"] = img
 
         ratio = min(self.test_size[0] / img.shape[0], self.test_size[1] / img.shape[1])
-        pad = [0,0,0,0]
+        pad = torch.tensor([0,0,0,0])
         pad[1] = ( self.test_size[0] - img.shape[0] * ratio ) / 2
         pad[0] = ( self.test_size[1] - img.shape[1] * ratio ) / 2
         pad[2] = pad[0]