Cast image to float32 (#702)

2023-03-14 08:13:19 -07:00
parent 54f499b623
commit 5e820818e9
1 changed files with 1 additions and 1 deletions
@@ -45,7 +45,7 @@ def infer(model, img):
  """

  # run the net
-  out = _infer(model, Tensor(img)).numpy()
+  out = _infer(model, Tensor(img.astype("float32"))).numpy()

  # if you want to look at the outputs
  """