vision-des 0.0.2

Dynamic Ensemble Selection framework for image classification

VisionDES

Robust and Explainable Dynamic Vision Ensemble

Installation

pip install vision-des

Quick Start

Create a Pool of Models

pool = [
    resnet50,
    efficientnet,
    convnext
]

Initialize VisionDES

from vision_des import VisionDES

des = VisionDES(
    dsel_dataset=dsel_dataset,
    pool=pool,
    device="cuda"
)

Build the Retrieval Index

des.fit()

Predict

prediction = des.predict(
    test_image,
    k=7
)

---

Explainable Inference

prediction = des.predict(
    test_image,
    k=7,
    explain=True
)