Views
No views yet
microsoft/swin-small-patch4-window7-224 para clasificar defectos en cartas Pokémon TCG.| Cabeza | Clases |
|---|---|
surface_scratch | none / micro / light / medium / heavy |
corner_wear | none / light / medium / heavy |
edge_whitening | none / light / medium / heavy |
| Métrica | Valor |
|---|---|
| val_loss | 0.1163 |
| Exactitud surface | 93.3% |
| Exactitud corners | 96.0% |
| Exactitud edges | 99.8% |
| Exactitud media | 96.4% |
1import onnxruntime as ort
2import numpy as np
3
4sess = ort.InferenceSession('card_defect_classifier.onnx')
5# img: np.array float32, shape (1, 3, 224, 224), normalizado ImageNet
6surface, corners, edges = sess.run(None, {'pixel_values': img})['none','micro','light','medium','heavy'] / ['none','light','medium','heavy'] / ['none','light','medium','heavy']