Upload folder using huggingface_hub
Browse files- LICENSE +0 -6
- README.md +11 -11
- config.json +5 -5
- example_infer.py +0 -1
- model.onnx +2 -2
LICENSE
CHANGED
|
@@ -4,10 +4,4 @@ Copyright (c) 2026 Easy4K
|
|
| 4 |
本模型权重与示例代码采用 Creative Commons Attribution-NonCommercial 4.0
|
| 5 |
International License(CC BY-NC 4.0)授权。
|
| 6 |
|
| 7 |
-
你可以自由地共享(复制、分发)与演绎(修改、转换、基于本作品创作),
|
| 8 |
-
但必须遵守以下条件:
|
| 9 |
-
- 署名:保留作者署名与许可证声明,并注明是否作出修改;
|
| 10 |
-
- 非商业性使用:不得将本作品用于商业目的。
|
| 11 |
-
|
| 12 |
许可证全文:https://creativecommons.org/licenses/by-nc/4.0/legalcode
|
| 13 |
-
中文摘要:https://creativecommons.org/licenses/by-nc/4.0/deed.zh-hans
|
|
|
|
| 4 |
本模型权重与示例代码采用 Creative Commons Attribution-NonCommercial 4.0
|
| 5 |
International License(CC BY-NC 4.0)授权。
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
许可证全文:https://creativecommons.org/licenses/by-nc/4.0/legalcode
|
|
|
README.md
CHANGED
|
@@ -89,11 +89,11 @@ print(float(score)) # 1.0 表示两帧重复
|
|
| 89 |
|
| 90 |
| 项目 | 数值 |
|
| 91 |
|---|---|
|
| 92 |
-
| 素材来源 | Wikimedia Commons(CC BY / CC BY-SA / CC0)
|
| 93 |
-
| 片段构造 | 固定窗口连续帧,按 1拍1 / 1拍2 / 1拍3 / 混合节奏展开,共
|
| 94 |
-
| 样本总数 |
|
| 95 |
| 正负比 | 1 : 1.00 |
|
| 96 |
-
| 难例 |
|
| 97 |
| 划分方式 | 按来源视频隔离,验证集素材不参与训练 |
|
| 98 |
|
| 99 |
CC 素材清单、作者与许可证见仓库内 `sources.json`。
|
|
@@ -106,20 +106,20 @@ CC 素材清单、作者与许可证见仓库内 `sources.json`。
|
|
| 106 |
| 优化器 | AdamW,weight decay 1e-4 |
|
| 107 |
| 学习率 | 2e-3,2 epoch 线性 warmup + 余弦退火 |
|
| 108 |
| Batch size | 256 |
|
| 109 |
-
| Epoch |
|
| 110 |
| 硬件 | NVIDIA P106-100(6 GB),CUDA 12.4,PyTorch 2.6.0 |
|
| 111 |
|
| 112 |
## 评测结果
|
| 113 |
|
| 114 |
-
### 与教师的一致性(验证集
|
| 115 |
|
| 116 |
| 指标 | 数值 |
|
| 117 |
|---|---|
|
| 118 |
-
| 软标签 MAE | 0.
|
| 119 |
-
| 判决准确率(阈值 0.5) | 97.
|
| 120 |
-
| 精确率 / 召回率 |
|
| 121 |
-
| 正样本 MAE | 0.
|
| 122 |
-
| 负样本 MAE | 0.
|
| 123 |
|
| 124 |
### 推理性能
|
| 125 |
|
|
|
|
| 89 |
|
| 90 |
| 项目 | 数值 |
|
| 91 |
|---|---|
|
| 92 |
+
| 素材来源 | Wikimedia Commons(CC BY / CC BY-SA / CC0);用户提供的术力口 / 真人 MV;公开网络 MV(术力口 / 动画 / 真人 / Live,仅训练使用,不随模型分发) |
|
| 93 |
+
| 片段构造 | 固定窗口连续帧,按 1拍1 / 1拍2 / 1拍3 / 混合节奏展开,共 5212 个片段 |
|
| 94 |
+
| 样本总数 | 550158(训练 472011 / 验证 78147) |
|
| 95 |
| 正负比 | 1 : 1.00 |
|
| 96 |
+
| 难例 | 9972 个(光流救回,权重 2.0) |
|
| 97 |
| 划分方式 | 按来源视频隔离,验证集素材不参与训练 |
|
| 98 |
|
| 99 |
CC 素材清单、作者与许可证见仓库内 `sources.json`。
|
|
|
|
| 106 |
| 优化器 | AdamW,weight decay 1e-4 |
|
| 107 |
| 学习率 | 2e-3,2 epoch 线性 warmup + 余弦退火 |
|
| 108 |
| Batch size | 256 |
|
| 109 |
+
| Epoch | 8 |
|
| 110 |
| 硬件 | NVIDIA P106-100(6 GB),CUDA 12.4,PyTorch 2.6.0 |
|
| 111 |
|
| 112 |
## 评测结果
|
| 113 |
|
| 114 |
+
### 与教师的一致性(验证集 78147 样本,按视频隔离)
|
| 115 |
|
| 116 |
| 指标 | 数值 |
|
| 117 |
|---|---|
|
| 118 |
+
| 软标签 MAE | 0.02788 |
|
| 119 |
+
| 判决准确率(阈值 0.5) | 97.56% |
|
| 120 |
+
| 精确率 / 召回率 | n/a / n/a |
|
| 121 |
+
| 正样本 MAE | 0.01063 |
|
| 122 |
+
| 负样本 MAE | 0.04085 |
|
| 123 |
|
| 124 |
### 推理性能
|
| 125 |
|
config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"model_name": "Easy4K SFDP - Siamese NET AI",
|
| 3 |
-
"model_version": "0.
|
| 4 |
"architecture": "siamese depthwise-separable CNN + difference/product fusion head",
|
| 5 |
"input": {
|
| 6 |
"names": [
|
|
@@ -35,10 +35,10 @@
|
|
| 35 |
],
|
| 36 |
"recommended_threshold": 0.5,
|
| 37 |
"metrics": {
|
| 38 |
-
"val_mae": 0.
|
| 39 |
-
"val_acc": 0.
|
| 40 |
-
"precision": 0.
|
| 41 |
-
"recall": 0.
|
| 42 |
},
|
| 43 |
"license": "cc-by-nc-4.0"
|
| 44 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"model_name": "Easy4K SFDP - Siamese NET AI",
|
| 3 |
+
"model_version": "0.5.0",
|
| 4 |
"architecture": "siamese depthwise-separable CNN + difference/product fusion head",
|
| 5 |
"input": {
|
| 6 |
"names": [
|
|
|
|
| 35 |
],
|
| 36 |
"recommended_threshold": 0.5,
|
| 37 |
"metrics": {
|
| 38 |
+
"val_mae": 0.02788,
|
| 39 |
+
"val_acc": 0.97562,
|
| 40 |
+
"precision": 0.0,
|
| 41 |
+
"recall": 0.0
|
| 42 |
},
|
| 43 |
"license": "cc-by-nc-4.0"
|
| 44 |
}
|
example_infer.py
CHANGED
|
@@ -9,7 +9,6 @@ def make_session(model_path="model.onnx"):
|
|
| 9 |
|
| 10 |
|
| 11 |
def score(sess, gray_a, gray_b):
|
| 12 |
-
"""gray_a / gray_b: 128x128 灰度块(uint8 或 float)。返回 0~1。"""
|
| 13 |
def prep(x):
|
| 14 |
x = np.asarray(x, dtype=np.float32)
|
| 15 |
if x.max() > 1.5:
|
|
|
|
| 9 |
|
| 10 |
|
| 11 |
def score(sess, gray_a, gray_b):
|
|
|
|
| 12 |
def prep(x):
|
| 13 |
x = np.asarray(x, dtype=np.float32)
|
| 14 |
if x.max() > 1.5:
|
model.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:72441f33662cc446fb79b1c3b9810e492ff4b071cf64aaa02fb0a7bdbe9dbb61
|
| 3 |
+
size 10431369
|