Skip to content

[Bug]: ValueError while exporting model in OpenVINO format with PTQ and ACQ #2592

Open
@srikesh-07

Description

@srikesh-07

Describe the bug

Hello,

I have tried to export the pre-trained PADiM model in OpenVINO format with Post Training Quantization (PTQ) and Accuracy Control Quantization (ACQ).
During execution, I got a ValueError that says 'ImageBatch' object is not subscriptable.

Dataset

MVTecAD

Model

PADiM

Steps to reproduce the behavior

  1. Install anomalib using pip.
  2. Execute the following command,
AnomalibCLI(
    args=[
        "export",
        "--export_type",
        ExportType.OPENVINO,
        "--compression_type",
        CompressionType.INT8_PTQ,
        "--data",
        MVTec,
        "--data.root",
        f"{project_path}/Padim/MVTec
        "--data.category",
        "dummy",
        "--model",
        "Padim",
        "--default_root_dir",
        str(project_path),
        "--trainer.max_epochs",
        "1",
        "--ckpt_path",
        f"{project_path}/Padim/MVTec/dummy/v0/weights/lightning/model.ckpt",
    ],
)

OS information

OS information:

  • OS: [e.g. Ubuntu 20.04]
  • Python version: [e.g. 3.10.0]
  • Anomalib version: [e.g. 0.3.6]
  • PyTorch version: [e.g. 1.9.0]
  • CUDA/cuDNN version: [e.g. 11.1]
  • GPU models and configuration: [e.g. 2x GeForce RTX 3090]
  • Any other relevant information: [e.g. I'm using a custom dataset]

Expected behavior

The model should be converted to OpenVINO format with PTQ which can be verified by successful execution.

Screenshots

No response

Pip/GitHub

pip

What version/branch did you use?

main

Configuration YAML

No Configuration YAML since AnomaliB CLI have been used.

Logs

The error arises from `models/components/base/export_mixin.py`

Code of Conduct

  • I agree to follow this project's Code of Conduct

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions