subnet_bridge: copy winning miner repo into library
Browse files- README.md +5 -5
- chute_config.yml +1 -1
- miner.py +3 -3
- pyproject.toml +1 -1
- 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:
|
| 17 |
-
winner_revision:
|
| 18 |
-
note: E=0.
|
| 19 |
---
|
| 20 |
|
| 21 |
# Detect-crime-winner
|
| 22 |
|
| 23 |
Published winning miner converted into a library element.
|
| 24 |
|
| 25 |
-
- Source winner repo: `
|
| 26 |
-
- Source revision: `
|
| 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
|
| 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", "
|
| 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.
|
| 54 |
)
|
| 55 |
_bonus_array = np.array(
|
| 56 |
-
[0.
|
| 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
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fe133b628068ddf780a7e94f9bbee73ff8abcd1b4dc1cc2f866b8bb2eeb14229
|
| 3 |
+
size 9809123
|