MTerryJack commited on
Commit
6edbc1a
·
verified ·
1 Parent(s): d0fbfad

subnet_bridge: copy winning miner repo into library

Browse files
Files changed (5) hide show
  1. README.md +5 -5
  2. chute_config.yml +1 -1
  3. miner.py +3 -3
  4. pyproject.toml +1 -1
  5. weights.onnx +2 -2
README.md CHANGED
@@ -13,17 +13,17 @@ tags:
13
  manako:
14
  source: winner_fetch
15
  manifest_element_name: manak0/Detect-crime
16
- winner_repo_id: thomaskk2024/pmodel_630
17
- winner_revision: 89943ba2d8face20c570c1d9bfa77dc8310aa3ea
18
- note: E=0.11999508 (map50=0.600000, size_mb=5.000205)
19
  ---
20
 
21
  # Detect-crime-winner
22
 
23
  Published winning miner converted into a library element.
24
 
25
- - Source winner repo: `thomaskk2024/pmodel_630`
26
- - Source revision: `89943ba2d8face20c570c1d9bfa77dc8310aa3ea`
27
  - Manifest element: `manak0/Detect-crime`
28
  - Element type: `Detect`
29
  - Objects: balaclava, hoodie, glove, bat, spray paint, graffiti
 
13
  manako:
14
  source: winner_fetch
15
  manifest_element_name: manak0/Detect-crime
16
+ winner_repo_id: carljurk/ScoreVision
17
+ winner_revision: 3b0a2a7e45ad080e10ea4219277642640ff78c39
18
+ note: E=0.06100469 (map50=0.600000, size_mb=9.835309)
19
  ---
20
 
21
  # Detect-crime-winner
22
 
23
  Published winning miner converted into a library element.
24
 
25
+ - Source winner repo: `carljurk/ScoreVision`
26
+ - Source revision: `3b0a2a7e45ad080e10ea4219277642640ff78c39`
27
  - Manifest element: `manak0/Detect-crime`
28
  - Element type: `Detect`
29
  - Objects: balaclava, hoodie, glove, bat, spray paint, graffiti
chute_config.yml CHANGED
@@ -2,7 +2,7 @@ Image:
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
- - pip install 'numpy>=1.23' 'onnxruntime-gpu>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
6
  - pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
7
 
8
  NodeSelector:
 
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
+ - pip install 'numpy>=1.23' 'onnxruntime-gpu==1.20.1' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
6
  - pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
7
 
8
  NodeSelector:
miner.py CHANGED
@@ -34,7 +34,7 @@ class Miner:
34
  # (ships inside weights.onnx), so a retrained model with a different class
35
  # order is remapped correctly without code changes. This static list is used
36
  # only when that metadata is missing or unparsable.
37
- model_class_names = ["balaclava", "bat", "glove", "graffiti", "hoodie", "spray paint"]
38
 
39
  input_size = 1280
40
 
@@ -50,10 +50,10 @@ class Miner:
50
  max_det = 150
51
 
52
  _conf_thres_array = np.array(
53
- [0.38, 0.58, 0.22, 0.25, 0.23, 0.22], dtype=np.float32,
54
  )
55
  _bonus_array = np.array(
56
- [0.18, 0.25, 0.12, 0.09, 0.07, 0.10], dtype=np.float32,
57
  )
58
 
59
  def __init__(self, path_hf_repo: Path) -> None:
 
34
  # (ships inside weights.onnx), so a retrained model with a different class
35
  # order is remapped correctly without code changes. This static list is used
36
  # only when that metadata is missing or unparsable.
37
+ model_class_names = ["balaclava", "hoodie", "glove", "bat", "spray paint", "graffiti"]
38
 
39
  input_size = 1280
40
 
 
50
  max_det = 150
51
 
52
  _conf_thres_array = np.array(
53
+ [0.33, 0.53, 0.32, 0.29, 0.46, 0.34], dtype=np.float32,
54
  )
55
  _bonus_array = np.array(
56
+ [0.02, 0.03, 0.04, 0.07, 0.06, 0.05], dtype=np.float32,
57
  )
58
 
59
  def __init__(self, path_hf_repo: Path) -> None:
pyproject.toml CHANGED
@@ -5,7 +5,7 @@ requires-python = ">=3.9"
5
 
6
  dependencies = [
7
  "numpy>=1.23",
8
- "onnxruntime>=1.16",
9
  "opencv-python>=4.7",
10
  "pillow>=9.5",
11
  "huggingface_hub>=0.19.4",
 
5
 
6
  dependencies = [
7
  "numpy>=1.23",
8
+ "onnxruntime==1.20.1",
9
  "opencv-python>=4.7",
10
  "pillow>=9.5",
11
  "huggingface_hub>=0.19.4",
weights.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:230d840c86bfde7e7b0ecff6d8114ebe88cfe67a8e6e51bce757075053470321
3
- size 4974021
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe133b628068ddf780a7e94f9bbee73ff8abcd1b4dc1cc2f866b8bb2eeb14229
3
+ size 9809123