concept_detection

Description

Model for concept detection, and its training and evaluation handle. See ConceptDetectionModel2D and ConceptDetection2DTrainTestHandle for details.

Classes

ConceptDetection2DTrainTestHandle	Train test handle for concept localization models.
ConceptDetectionModel2D	Pytorch model implementation of a concept embedding for 2D conv layers.