search

NKI-AI / ahcore

Ahcore is the AI for Oncology core computational pathology toolkit
Apache License 2.0
15 stars 1 forks source link

AhcoreJitModel extended for specific jit compiled inference use cases like extracting features from foundation models #82

Closed AjeyPaiK closed 3 months ago

AjeyPaiK commented 3 months ago

Fixes #81

  1. AhcoreJitModel can now be inherited based on the type of the JIT model being used.
  2. A class FoundationModel is inherited from AhcoreJitModel which can be used for feature extraction (based on large models like Dinov2.)
  3. A class SegmentationModel is inherited from AhcoreJitModel that can be used for segmentation inference.
  4. In ahcore.utils.types, several custom types are defined so that ahcore can easily be extended to any type of JIT compiled models.