ONNX 模型库
返回模型

说明文档

JunkWaxHero 🦸‍♂️

JunkWaxHero 是您识别 1980 至 1999 年"垃圾蜡"棒球卡套装黄金时代的首选计算机视觉模型。无论您是收藏家、经销商还是爱好者,JunkWaxHero 都能以高精确率和召回率简化您的收藏编目和验证流程。JunkWaxHero 的目标是提供一个无需依赖基于 SaaS 的推理 API 即可快速识别棒球卡套装的模型。

📦 可用格式

JunkWaxHero 提供 ONNX 格式,以满足不同的部署需求和偏好。

  • ONNX: 非常适合跨各种平台的互操作性,并针对不同的硬件加速器进行优化。

📊 训练详情

  • 模型版本: 第 24 次迭代
  • 领域: 通用(紧凑型)[S1]

性能指标 📊

  • 精确率: 98.3%
  • 召回率: 95.6%
  • mAP: 98.6%

各标签性能 🏷️

标签 精确率 召回率 平均精度 (AP)
1982 Donruss 100.0% 100.0% 100.0%
1984 Topps 100.0% 100.0% 100.0%
1987 Fleer 100.0% 95.5% 95.5%
1987 Topps 100.0% 100.0% 100.0%
1988 Donruss 91.3% 100.0% 100.0%
1988 Fleer 100.0% 76.2% 95.2%
1988 Fleer Pack 100.0% 87.5% 100.0%
1988 Score 100.0% 100.0% 100.0%
1988 Topps 96.0% 100.0% 100.0%
1988 Topps Pack 100.0% 100.0% 100.0%
1989 Bowman 100.0% 94.7% 100.0%
1989 Donruss 100.0% 94.7% 100.0%
1989 Fleer 100.0% 100.0% 100.0%
1989 Score 100.0% 100.0% 100.0%
1989 Topps 95.0% 90.5% 90.5%
1989 Topps Pack 100.0% 83.3% 91.0%
1989 Upper Deck 100.0% 92.3% 100.0%
1990 Donruss 100.0% 100.0% 100.0%
1990 Fleer 88.6% 96.9% 95.7%
1990 Fleer Pack 100.0% 100.0% 100.0%
1990 Leaf 100.0% 100.0% 100.0%
1990 Leaf Pack 100.0% 80.0% 100.0%
1990 Topps 100.0% 100.0% 100.0%
1990 Upper Deck High Series Pack 100.0% 91.3% 100.0%
1991 Fleer Ultra 100.0% 97.5% 100.0%
1991 Leaf 100.0% 90.9% 95.5%
1991 Leaf Pack 100.0% 75.0% 100.0%
1991 Topps 92.9% 81.3% 97.2%
1991 Topps Pack 100.0% 91.7% 100.0%
1991 Upper Deck 100.0% 89.5% 94.7%
1991 Upper Deck Low Series Pack 100.0% 96.0% 98.9%
1992 Fleer 100.0% 100.0% 100.0%
1992 Fleer Ultra 100.0% 100.0% 100.0%
1992 Fleer Pack 100.0% 50.0% 91.7%
1992 O-Pee-Chee Premiere 100.0% 94.7% 99.7%
1992 Pinnacle 100.0% 94.4% 99.7%
1992 Pinnacle Pack 100.0% 85.7% 100.0%
1992 Upper Deck 95.5% 87.5% 87.5%
1992 Upper Deck High Series Pack 100.0% 100.0% 100.0%
1993 Fleer 100.0% 95.0% 99.8%
1993 Fleer Series 1 Pack 100.0% 100.0% 100.0%
1993 Fleer Series 2 Pack 95.0% 100.0% 100.0%
1993 Topps 86.4% 86.4% 98.1%
1994 Leaf 95.7% 100.0% 100.0%
1994 Pinnacle 100.0% 100.0% 100.0%
1994 Score 100.0% 100.0% 100.0%
1995 Leaf 100.0% 100.0% 100.0%
1995 Select 95.5% 100.0% 100.0%
1996 Pinnacle 100.0% 100.0% 100.0%

🚀 快速开始

仓库

在 Hugging Face 上访问仓库:https://huggingface.co/enusbaum/JunkWaxHero/

安装

要将 JunkWaxHero 集成到您的项目中,请选择最适合您工作流程的格式:

ONNX

  1. 安装 ONNX Runtime:

    pip install onnxruntime
    
  2. 下载 ONNX 模型:

    导航到仓库中的 ONNX 目录,从 https://huggingface.co/enusbaum/JunkWaxHero/tree/main/onnx 下载 model.onnx 文件。

  3. 加载 ONNX 模型:

    import onnxruntime as ort
    import numpy as np
    
    # Path to the downloaded ONNX model
    MODEL_PATH = 'path/to/junkwaxhero/onnx/junkwaxhero.onnx'
    
    # Initialize the ONNX runtime session
    session = ort.InferenceSession(MODEL_PATH)
    
    # Get input and output names
    input_name = session.get_inputs()[0].name
    output_name = session.get_outputs()[0].name
    
    # Example function to run inference
    def predict(image):
        image = np.expand_dims(image, axis=0).astype(np.float32)  # Add batch dimension and ensure correct type
        predictions = session.run([output_name], {input_name: image})
        return predictions
    

使用

  1. 准备图像: 确保您的棒球卡图像清晰且光线充足,以获得最佳识别效果。图像应为 4:3 比例垂直对齐

  2. 加载模型: 根据您选择的格式(TensorFlow 或 ONNX),使用上述相应的代码片段。

  3. 识别套装: 将您的图像通过 JunkWaxHero 进行处理,以准确识别它们所属的套装。

    from PIL import Image
    
    # Load and preprocess the image
    def load_image(image_path):
        image = Image.open(image_path)
        image = image.resize((width, height))  # Resize to the input size expected by the model
        image = np.array(image)
        return image
    
    # Example usage
    image = load_image('path/to/baseball_card.jpg')
    predictions = predict(image)
    
    # Get the predicted label
    predicted_label = labels[np.argmax(predictions)]
    print(f'Predicted Set: {predicted_label}')
    

    注意: 请将 (width, height) 替换为您的模型实际期望的尺寸。

优化

利用高精确率和召回率来维护一个有条理且经过验证的收藏。根据您的部署环境,选择 TensorFlow 以与 TensorFlow 生态系统无缝集成,或选择 ONNX 以获得更广泛的平台兼容性和性能优化。

🔍 使用场景

  • 收藏家: 快速验证和编目您的棒球卡收藏。
  • 经销商: 通过准确的套装识别简化库存管理。
  • 爱好者: 使用先进的计算机视觉技术轻松探索和发现各种套装。

🛠️ 技术详情

JunkWaxHero 采用最先进的计算机视觉架构,专门优化用于从图像中识别棒球卡套装。该模型在跨越多个十年的多样化数据集上进行了训练,确保在各种卡片状况和设计下具有鲁棒性和可靠性。通过同时提供 TensorFlow 和 ONNX 格式,JunkWaxHero 为不同的部署场景提供了灵活性,无论您是集成到现有的 TensorFlow 流程中,还是通过 ONNX 寻求跨平台兼容性。

📄 许可证

本项目采用 MIT 许可证授权。

📫 联系方式

如有问题、建议或需要支持,请联系 eric@nusbaum.me


收藏愉快!

enusbaum/JunkWaxHero

作者 enusbaum

object-detection
↓ 0 ♥ 0

创建时间: 2025-01-10 23:17:52+00:00

更新时间: 2025-01-28 01:56:55+00:00

在 Hugging Face 上查看

文件 (8)

.gitattributes
README.md
onnx/LICENSE
onnx/README.md
onnx/cvexport.manifest
onnx/labels.txt
onnx/metadata_properties.json
onnx/model.onnx ONNX