thomaskk2024 commited on
Commit
945aea0
·
verified ·
1 Parent(s): f4bb19a

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. miner.py +2 -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.20, 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-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:0f0c937da3ee892f586a42b3a808d90f44e5c0f1ccc51c19d4889f5536a2b816
3
- size 9844493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1acb453d39ccab0d21164a05c5b5ee27b4406981e7aede31fecf05fc673ec0c
3
+ size 9809208