meaculpitt commited on
Commit
418b52f
·
verified ·
1 Parent(s): bd99c2f

deploy push for beverage (deploy)

Browse files
Files changed (1) hide show
  1. miner.py +3 -1
miner.py CHANGED
@@ -148,7 +148,9 @@ class Miner:
148
  canvas, ratio, pad = self._letterbox(image_bgr)
149
  rgb = cv2.cvtColor(canvas, cv2.COLOR_BGR2RGB)
150
  x = (rgb.astype(np.float32) / 255.0).transpose(2, 0, 1)[None, ...]
151
- return np.ascontiguousarray(x, dtype=np.float32), ratio, pad
 
 
152
 
153
  # ---------------------------------------------------------------- nms helpers
154
  @staticmethod
 
148
  canvas, ratio, pad = self._letterbox(image_bgr)
149
  rgb = cv2.cvtColor(canvas, cv2.COLOR_BGR2RGB)
150
  x = (rgb.astype(np.float32) / 255.0).transpose(2, 0, 1)[None, ...]
151
+ # v8 weights are FP16 — input dtype must match or onnxruntime errors with
152
+ # "Unexpected input data type. Actual: float, expected: float16".
153
+ return np.ascontiguousarray(x, dtype=np.float16), ratio, pad
154
 
155
  # ---------------------------------------------------------------- nms helpers
156
  @staticmethod