DraconicDragon commited on
Commit
267328b
·
1 Parent(s): 4e5ec04

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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
- - `lsnet_xl_artist-dynamo-opset18_merged.onnx`: 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,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
  **免责声明**: 本模型仅供研究和教育用途。在商业应用中使用时,请确保遵守相关法律法规和伦理准则。