Commit ·
267328b
1
Parent(s): 4e5ec04
Update README.md
Browse files
README.md
CHANGED
|
@@ -10,7 +10,7 @@ Experimental onnx conversion of kaloscope model.
|
|
| 10 |
[🤗 Space Demo](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier) for ONNX & PyTorch inference implementation (incl. timm+lsnet; OpenVINO accelerate CPU inference; no Triton required - refer to [ska.py](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier/blob/main/lsnet/ska.py) or [here](https://github.com/spawner1145/comfyui-lsnet/pull/2))
|
| 11 |
Barebones ONNX inference script (no timm or lsnet; scores are a tiny bit different - probably different img preprocessing): [onnx_barebones_inference.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/onnx_barebones_inference.py)
|
| 12 |
|
| 13 |
-
- `
|
| 14 |
- 'merged' in name because torch.onnx.export gave me 2 files - .onnx.data and a small .onnx file - I used [merge_onnx_ext_data.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/merge_onnx_ext_data.py) to merge them (`external_data=False` can be set for torch.onnx.export I think but didn't use it)
|
| 15 |
|
| 16 |
Conversion script used: [kaloscope_pth2onnx.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/convert_scripts/kaloscope_pth2onnx.py) | Related Info: [README.md](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/convert_scripts/README.md)
|
|
@@ -100,11 +100,9 @@ pip install torch torchvision timm
|
|
| 100 |
```python
|
| 101 |
import torch
|
| 102 |
from timm.models import create_model
|
| 103 |
-
|
| 104 |
# 加载模型
|
| 105 |
model = create_model('lsnet_t_artist', pretrained=True, num_classes=31770)
|
| 106 |
model.eval()
|
| 107 |
-
|
| 108 |
# 推理
|
| 109 |
with torch.no_grad():
|
| 110 |
output = model(input_tensor)
|
|
@@ -138,7 +136,9 @@ with torch.no_grad():
|
|
| 138 |
- 基于Danbooru数据集训练
|
| 139 |
- 支持31,770个艺术家类别
|
| 140 |
- 达到84.2%的分类准确率
|
| 141 |
-
|
|
|
|
|
|
|
| 142 |
---
|
| 143 |
|
| 144 |
**免责声明**: 本模型仅供研究和教育用途。在商业应用中使用时,请确保遵守相关法律法规和伦理准则。
|
|
|
|
| 10 |
[🤗 Space Demo](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier) for ONNX & PyTorch inference implementation (incl. timm+lsnet; OpenVINO accelerate CPU inference; no Triton required - refer to [ska.py](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier/blob/main/lsnet/ska.py) or [here](https://github.com/spawner1145/comfyui-lsnet/pull/2))
|
| 11 |
Barebones ONNX inference script (no timm or lsnet; scores are a tiny bit different - probably different img preprocessing): [onnx_barebones_inference.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/onnx_barebones_inference.py)
|
| 12 |
|
| 13 |
+
- `kaloscope_1-0.onnx`: Exported from `best_checkpoint.pth` original Kaloscope release | dynamo=True, dynamic_axes=None, opset_version=None (torch 2.8.0 used here defaults to 18 when None), optimization/constant folding enabled
|
| 14 |
- 'merged' in name because torch.onnx.export gave me 2 files - .onnx.data and a small .onnx file - I used [merge_onnx_ext_data.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/merge_onnx_ext_data.py) to merge them (`external_data=False` can be set for torch.onnx.export I think but didn't use it)
|
| 15 |
|
| 16 |
Conversion script used: [kaloscope_pth2onnx.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/convert_scripts/kaloscope_pth2onnx.py) | Related Info: [README.md](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/convert_scripts/README.md)
|
|
|
|
| 100 |
```python
|
| 101 |
import torch
|
| 102 |
from timm.models import create_model
|
|
|
|
| 103 |
# 加载模型
|
| 104 |
model = create_model('lsnet_t_artist', pretrained=True, num_classes=31770)
|
| 105 |
model.eval()
|
|
|
|
| 106 |
# 推理
|
| 107 |
with torch.no_grad():
|
| 108 |
output = model(input_tensor)
|
|
|
|
| 136 |
- 基于Danbooru数据集训练
|
| 137 |
- 支持31,770个艺术家类别
|
| 138 |
- 达到84.2%的分类准确率
|
| 139 |
+
### v1.1 (2025年10月)
|
| 140 |
+
- 150epoch
|
| 141 |
+
- 达到85.6%的分类准确率
|
| 142 |
---
|
| 143 |
|
| 144 |
**免责声明**: 本模型仅供研究和教育用途。在商业应用中使用时,请确保遵守相关法律法规和伦理准则。
|