Upload folder using huggingface_hub
Browse files- miner.py +2 -2
- weights.onnx +2 -2
miner.py
CHANGED
|
@@ -59,15 +59,15 @@ class Miner:
|
|
| 59 |
min_box_size = 2.0
|
| 60 |
|
| 61 |
_conf_thres_array = np.array(
|
| 62 |
-
[0.38, 0.38, 0.22, 0.
|
| 63 |
)
|
| 64 |
_bonus_array = np.array(
|
| 65 |
[0.2, 0.25, 0.12, 0.09, 0.21, 0.06], dtype=np.float32,
|
| 66 |
)
|
| 67 |
|
| 68 |
def __init__(self, path_hf_repo: Path) -> None:
|
| 69 |
-
# model_path = path_hf_repo / "weights-v17-np2-704-0730.onnx"
|
| 70 |
model_path = path_hf_repo / "weights.onnx"
|
|
|
|
| 71 |
print("ORT version:", ort.__version__)
|
| 72 |
|
| 73 |
try:
|
|
|
|
| 59 |
min_box_size = 2.0
|
| 60 |
|
| 61 |
_conf_thres_array = np.array(
|
| 62 |
+
[0.38, 0.38, 0.22, 0.22, 0.33, 0.20], dtype=np.float32,
|
| 63 |
)
|
| 64 |
_bonus_array = np.array(
|
| 65 |
[0.2, 0.25, 0.12, 0.09, 0.21, 0.06], dtype=np.float32,
|
| 66 |
)
|
| 67 |
|
| 68 |
def __init__(self, path_hf_repo: Path) -> None:
|
|
|
|
| 69 |
model_path = path_hf_repo / "weights.onnx"
|
| 70 |
+
# model_path = path_hf_repo / "weights.onnx"
|
| 71 |
print("ORT version:", ort.__version__)
|
| 72 |
|
| 73 |
try:
|
weights.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d1acb453d39ccab0d21164a05c5b5ee27b4406981e7aede31fecf05fc673ec0c
|
| 3 |
+
size 9809208
|