Commit ·
6f43399
1
Parent(s): 07302a4
Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,10 @@ also extra pytorch ema only version (`best_checkpoint_ema.pth`) of the model whi
|
|
| 11 |
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)
|
| 12 |
There's an ""issue"" with the onnx models: even when using timm+lsnet impl like the pytorch inference code does im getting correct results but with rather large confidence differences which might be expected because of the conversion and onnx optimization but 19%->27.8% seems a bit large
|
| 13 |
|
| 14 |
-
Huggingface space demo for ONNX inference implementation: https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
**model name info** (optimization/constant folding enabled for all of them)
|
| 17 |
|
|
@@ -26,8 +29,9 @@ Huggingface space demo for ONNX inference implementation: https://huggingface.co
|
|
| 26 |
Note that newer opset versions might not work with older tooling. I haven't tested this.
|
| 27 |
**Using opset 18 is probably the best choice** - I don't see any score difference between the different opset except opset 13 but that one also didn't have dynamo enabled during export and I assume dynamo is probably just better?
|
| 28 |
|
| 29 |
-
|
| 30 |
|
|
|
|
| 31 |
|
| 32 |
# LSNet 艺术家风格分类模型 Model Card
|
| 33 |
|
|
|
|
| 11 |
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)
|
| 12 |
There's an ""issue"" with the onnx models: even when using timm+lsnet impl like the pytorch inference code does im getting correct results but with rather large confidence differences which might be expected because of the conversion and onnx optimization but 19%->27.8% seems a bit large
|
| 13 |
|
| 14 |
+
Huggingface space demo for ONNX inference implementation (incl. timm+lsnet): https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier
|
| 15 |
+
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)
|
| 16 |
+
|
| 17 |
+
<hr>
|
| 18 |
|
| 19 |
**model name info** (optimization/constant folding enabled for all of them)
|
| 20 |
|
|
|
|
| 29 |
Note that newer opset versions might not work with older tooling. I haven't tested this.
|
| 30 |
**Using opset 18 is probably the best choice** - I don't see any score difference between the different opset except opset 13 but that one also didn't have dynamo enabled during export and I assume dynamo is probably just better?
|
| 31 |
|
| 32 |
+
<hr>
|
| 33 |
|
| 34 |
+
Original description:
|
| 35 |
|
| 36 |
# LSNet 艺术家风格分类模型 Model Card
|
| 37 |
|