deploy push for beverage (deploy)
Browse files
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 |
-
|
|
|
|
|
|
|
| 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
|