Closed AjeyPaiK closed 2 months ago
Hi @EricMarcus-ai, thanks for the great comments and suggestions to improve. Your reference codebase really helped me.
Two things after my latest push, The comments to which I haven't replied are no longer applicable because the code is changed. The ones that I've replied to have been addressed.
Another round of review should set this right.
Fixes #81
AhcoreJitModel
is renamed toBaseAhcoreJitModel
BaseAhcoreJitModel
can now be inherited based on the type of the JIT model being used.DinoV2JitModel
is inherited fromBaseAhcoreJitModel
which can be used for feature extraction (based on large models like Dinov2.)SegmentationJitModel
is inherited fromBaseAhcoreJitModel
that can be used for segmentation inference.ahcore.utils.types
, several custom types are defined so that ahcore can easily be extended to any type of JIT compiled models.