aai-institute / ProtoTree

Fork of ProtoTrees: Neural Prototype Trees for Interpretable Fine-grained Image Recognition, published at CVPR2021
Other
3 stars 1 forks source link

Prototypes explanation computed only for batch size = 1 #54

Open gpitteri opened 1 year ago

gpitteri commented 1 year ago

To evaluate the local scores of the prototypes we need to run inference on the dataset, hence create a dataloader. In the current implementation the dataloader needs to have a batch size of 1, otherwise we get an error.