reworked iop_corr, added explainability tools and plotting tools, cleanup codebase

2026-03-12 16:12:59 +01:00
parent 13ad32683f
commit 6d0698d0fb
35 changed files with 7765 additions and 5723 deletions
@@ -7,8 +7,8 @@ import torch
 from torch import nn
 from torchvision import transforms

-from classes.backbones import BACKBONES, list_names, load_backbone_weights
-from classes.SE_attention import SEBlock
+from classes.v2.backbones import BACKBONES, list_names, load_backbone_weights
+from classes.v2.SE_attention import SEBlock
 from classes.v2.data_bundle import DataBundle