quadra 0.0.1__py3-none-any.whl → 2.2.7__py3-none-any.whl

quadra/configs/model/classification.yaml

@@ -0,0 +1,10 @@
+model: ${backbone.model}
+
+pre_classifier: null
+classifier:
+  _target_: torch.nn.Linear
+  in_features: ${backbone.metadata.output_dim}
+module:
+  _target_: quadra.modules.classification.ClassificationModule
+  lr_scheduler_interval: "epoch"
+  criterion: ${loss}

quadra/configs/model/dino.yaml

@@ -0,0 +1,26 @@
+student_projection_mlp:
+  _target_: quadra.modules.ssl.common.DinoProjectionHead
+  input_dim: ${backbone.metadata.output_dim}
+  output_dim: 1024
+  hidden_dim: 256
+  bottleneck_dim: 128
+  num_layers: 3
+  norm_last_layer: True
+
+student: ${backbone.model}
+
+teacher_projection_mlp: ${model.student_projection_mlp}
+
+teacher: ${model.student}
+
+module:
+  _target_: quadra.modules.ssl.Dino
+  classifier: ${model.classifier}
+  criterion: ${loss}
+  teacher_momentum: 0.996
+  lr_scheduler_interval: step
+
+classifier:
+  _target_: sklearn.neighbors.KNeighborsClassifier
+  n_neighbors: 20
+  n_jobs: 5

quadra/configs/model/logistic_regression.yaml

@@ -0,0 +1,4 @@
+_target_: sklearn.linear_model.LogisticRegression
+max_iter: 10000
+random_state: ${core.seed}
+n_jobs: 10

quadra/configs/model/multilabel_classification.yaml

@@ -0,0 +1,9 @@
+model: ${backbone.model}
+classifier:
+  _target_: torch.nn.Linear
+  in_features: ${backbone.metadata.output_dim}
+  out_features: ${datamodule.num_classes}
+module:
+  _target_: quadra.modules.classification.MultilabelClassificationModule
+  lr_scheduler_interval: "epoch"
+  criterion: ${loss}

quadra/configs/model/simclr.yaml

@@ -0,0 +1,18 @@
+model: ${backbone.model}
+
+projection_mlp:
+  _target_: quadra.modules.ssl.common.SimCLRProjectionHead
+  input_dim: ${backbone.metadata.output_dim}
+  output_dim: ${backbone.metadata.output_dim}
+  hidden_dim: ${backbone.metadata.output_dim}
+
+module:
+  _target_: quadra.modules.ssl.SimCLR
+  classifier: ${model.classifier}
+  criterion: ${loss}
+  lr_scheduler_interval: step
+
+classifier:
+  _target_: sklearn.neighbors.KNeighborsClassifier
+  n_neighbors: 5
+  n_jobs: 5

quadra/configs/model/simsiam.yaml

@@ -0,0 +1,24 @@
+model: ${backbone.model}
+
+projection_mlp:
+  _target_: quadra.modules.ssl.common.SimSiamProjectionHead
+  input_dim: ${backbone.metadata.output_dim}
+  output_dim: ${backbone.metadata.output_dim}
+  hidden_dim: ${backbone.metadata.output_dim}
+
+prediction_mlp:
+  _target_: quadra.modules.ssl.common.SimSiamPredictionHead
+  input_dim: ${backbone.metadata.output_dim}
+  output_dim: ${backbone.metadata.output_dim}
+  hidden_dim: 128
+
+module:
+  _target_: quadra.modules.ssl.SimSIAM
+  classifier: ${model.classifier}
+  criterion: ${loss}
+  lr_scheduler_interval: step
+
+classifier:
+  _target_: sklearn.neighbors.KNeighborsClassifier
+  n_neighbors: 5
+  n_jobs: 5

quadra/configs/model/smp.yaml

@@ -0,0 +1,4 @@
+model: ${backbone.model}
+module:
+  _target_: quadra.modules.base.SegmentationModel
+  loss_fun: ${loss}

quadra/configs/model/smp_multiclass.yaml

@@ -0,0 +1,4 @@
+model: ${backbone.model}
+module:
+  _target_: quadra.modules.base.SegmentationModelMulticlass
+  loss_fun: ${loss}

quadra/configs/model/vicreg.yaml

@@ -0,0 +1,16 @@
+model: ${backbone.model}
+projection_mlp_mult: 4
+projection_mlp:
+  _target_: quadra.modules.ssl.common.BarlowTwinsProjectionHead
+  input_dim: ${backbone.metadata.output_dim}
+  output_dim: ${backbone.metadata.output_dim}
+  hidden_dim: ${backbone.metadata.output_dim}
+module:
+  _target_: quadra.modules.ssl.BarlowTwins
+  classifier: ${model.classifier}
+  criterion: ${loss}
+  lr_scheduler_interval: step
+classifier:
+  _target_: sklearn.neighbors.KNeighborsClassifier
+  n_neighbors: 20
+  n_jobs: 5

quadra/configs/optimizer/adam.yaml

@@ -0,0 +1,5 @@
+_target_: torch.optim.Adam
+lr: 0.001
+betas: [0.9, 0.999]
+eps: 1e-08
+weight_decay: 0

quadra/configs/optimizer/adamw.yaml

@@ -0,0 +1,3 @@
+_target_: torch.optim.AdamW
+lr: 0.01
+weight_decay: 0.01

quadra/configs/optimizer/default.yaml

@@ -0,0 +1,4 @@
+_target_: torch.optim.SGD
+lr: 0.01
+momentum: .9
+weight_decay: 0.0001

quadra/configs/optimizer/lars.yaml

@@ -0,0 +1,8 @@
+_target_: quadra.optimizers.LARS
+lr: 0.01
+momentum: 0.9
+dampening: 0
+weight_decay: 1e-5
+nesterov: true
+trust_coefficient: 0.001
+eps: 1e-8

quadra/configs/optimizer/sgd.yaml

@@ -0,0 +1,4 @@
+_target_: torch.optim.SGD
+lr: 0.01
+momentum: .9
+weight_decay: 0.0001

quadra/configs/scheduler/default.yaml

@@ -0,0 +1,5 @@
+_target_: torch.optim.lr_scheduler.ReduceLROnPlateau
+patience: 20
+factor: 0.5
+verbose: False
+threshold: 0.01

quadra/configs/scheduler/rop.yaml

@@ -0,0 +1,5 @@
+_target_: torch.optim.lr_scheduler.ReduceLROnPlateau
+patience: 20
+factor: 0.995
+verbose: False
+threshold: 0.01

quadra/configs/scheduler/step.yaml

@@ -0,0 +1,3 @@
+_target_: torch.optim.lr_scheduler.StepLR
+step_size: 30
+gamma: 0.5

quadra/configs/scheduler/warmrestart.yaml

@@ -0,0 +1,2 @@
+_target_: torch.optim.lr_scheduler.CosineAnnealingWarmRestarts
+T_0: ${trainer.max_epochs}

quadra/configs/scheduler/warmup.yaml

@@ -0,0 +1,6 @@
+_target_: quadra.schedulers.CosineAnnealingWithLinearWarmUp
+total_epochs: ${trainer.max_epochs}
+linear_warmup_epochs: 10
+scheduler_interval: ${model.module.lr_scheduler_interval}
+init_lr:
+  - 0.001

quadra/configs/task/anomalib/cfa.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.cfa.lightning_model.CfaLightning

quadra/configs/task/anomalib/cflow.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.cflow.lightning_model.CflowLightning

quadra/configs/task/anomalib/csflow.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.csflow.lightning_model.CsflowLightning

quadra/configs/task/anomalib/draem.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.draem.lightning_model.DraemLightning

quadra/configs/task/anomalib/efficient_ad.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.efficient_ad.lightning_model.EfficientAdLightning

quadra/configs/task/anomalib/fastflow.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.fastflow.lightning_model.FastflowLightning

quadra/configs/task/anomalib/inference.yaml

@@ -0,0 +1,3 @@
+_target_: quadra.tasks.anomaly.AnomalibEvaluation
+model_path: ???
+device:

quadra/configs/task/anomalib/padim.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.padim.lightning_model.PadimLightning

quadra/configs/task/anomalib/patchcore.yaml

@@ -0,0 +1,5 @@
+_target_: quadra.tasks.anomaly.AnomalibDetection
+report: true
+run_test: true
+module_function:
+  _target_: anomalib.models.patchcore.lightning_model.PatchcoreLightning

quadra/configs/task/classification.yaml

@@ -0,0 +1,6 @@
+_target_: quadra.tasks.Classification
+lr_multiplier: null
+output:
+  example: false
+report: false
+run_test: true

quadra/configs/task/classification_evaluation.yaml

@@ -0,0 +1,6 @@
+_target_: quadra.tasks.ClassificationEvaluation
+device: cuda:0
+output:
+  example: true
+model_path: ???
+report: true

quadra/configs/task/default.yaml

@@ -0,0 +1 @@
+_target_: quadra.tasks.PlaceholderTask

quadra/configs/task/segmentation.yaml

@@ -0,0 +1,9 @@
+_target_: quadra.tasks.Segmentation
+num_viz_samples: 10
+checkpoint_path: null
+run_test: true
+report: false
+evaluate:
+  analysis: false
+  control: false
+  corruption: false

quadra/configs/task/segmentation_evaluation.yaml

@@ -0,0 +1,3 @@
+_target_: quadra.tasks.SegmentationAnalysisEvaluation
+device: cuda:0
+model_path: ???

quadra/configs/task/sklearn_classification.yaml

@@ -0,0 +1,13 @@
+_target_: quadra.tasks.SklearnClassification
+device: cuda:0
+half_precision: false
+gradcam: false
+automatic_batch_size:
+  starting_batch_size: 1024
+  disable: true
+save_model_summary: false
+output:
+  folder: classification_experiment
+  report: true
+  example: true
+  test_full_data: true

quadra/configs/task/sklearn_classification_patch.yaml

@@ -0,0 +1,11 @@
+_target_: quadra.tasks.PatchSklearnClassification
+device: cuda:0
+half_precision: false
+automatic_batch_size:
+  starting_batch_size: 1024
+  disable: true
+output:
+  folder: classification_patch_experiment
+  report: true
+  example: true
+  reconstruction_method: major_voting

quadra/configs/task/sklearn_classification_patch_test.yaml

@@ -0,0 +1,8 @@
+_target_: quadra.tasks.PatchSklearnTestClassification
+device: cuda:0
+output:
+  folder: classification_patch_test
+  report: true
+  example: true
+  reconstruction_method: major_voting
+model_path: ???

quadra/configs/task/sklearn_classification_test.yaml

@@ -0,0 +1,8 @@
+_target_: quadra.tasks.classification.SklearnTestClassification
+device: cuda:0
+gradcam: false
+output:
+  folder: classification_experiment
+  report: true
+  example: true
+model_path: ???

quadra/configs/task/ssl.yaml

@@ -0,0 +1,2 @@
+_target_: quadra.tasks.SSL
+run_test: false

quadra/configs/trainer/lightning_cpu.yaml

@@ -0,0 +1,36 @@
+_target_: pytorch_lightning.Trainer
+devices: 1
+accelerator: cpu
+min_epochs: 1
+max_epochs: 10
+strategy: auto
+log_every_n_steps: 10
+precision: 32
+num_nodes: 1
+fast_dev_run: false
+max_time:
+limit_train_batches:
+limit_val_batches:
+limit_test_batches:
+limit_predict_batches:
+overfit_batches: 0.0
+val_check_interval:
+check_val_every_n_epoch: 1
+num_sanity_val_steps:
+enable_checkpointing:
+enable_progress_bar:
+enable_model_summary:
+accumulate_grad_batches: 1
+gradient_clip_val:
+gradient_clip_algorithm:
+deterministic:
+benchmark:
+inference_mode: True
+use_distributed_sampler: True
+profiler:
+detect_anomaly: False
+barebones: False
+plugins:
+sync_batchnorm: False
+reload_dataloaders_every_n_epochs: 0
+default_root_dir:

quadra/configs/trainer/lightning_gpu.yaml

@@ -0,0 +1,35 @@
+_target_: pytorch_lightning.Trainer
+devices: [0]
+accelerator: gpu
+min_epochs: 1
+max_epochs: 10
+log_every_n_steps: 10
+strategy: auto
+num_nodes: 1
+fast_dev_run: false
+max_time:
+limit_train_batches:
+limit_val_batches:
+limit_test_batches:
+limit_predict_batches:
+overfit_batches: 0.0
+val_check_interval:
+check_val_every_n_epoch: 1
+num_sanity_val_steps:
+enable_checkpointing:
+enable_progress_bar:
+enable_model_summary:
+accumulate_grad_batches: 1
+gradient_clip_val:
+gradient_clip_algorithm:
+deterministic:
+benchmark:
+inference_mode: True
+use_distributed_sampler: True
+profiler:
+detect_anomaly: False
+barebones: False
+plugins:
+sync_batchnorm: False
+reload_dataloaders_every_n_epochs: 0
+default_root_dir:

quadra/configs/trainer/lightning_gpu_bf16.yaml

@@ -0,0 +1,36 @@
+_target_: pytorch_lightning.Trainer
+accelerator: gpu
+devices: [0]
+min_epochs: 1
+max_epochs: 10
+log_every_n_steps: 10
+precision: bf16
+strategy: auto
+num_nodes: 1
+fast_dev_run: false
+max_time:
+limit_train_batches:
+limit_val_batches:
+limit_test_batches:
+limit_predict_batches:
+overfit_batches: 0.0
+val_check_interval:
+check_val_every_n_epoch: 1
+num_sanity_val_steps:
+enable_checkpointing:
+enable_progress_bar:
+enable_model_summary:
+accumulate_grad_batches: 1
+gradient_clip_val:
+gradient_clip_algorithm:
+deterministic:
+benchmark:
+inference_mode: True
+use_distributed_sampler: True
+profiler:
+detect_anomaly: False
+barebones: False
+plugins:
+sync_batchnorm: False
+reload_dataloaders_every_n_epochs: 0
+default_root_dir:

quadra/configs/trainer/lightning_gpu_fp16.yaml

@@ -0,0 +1,36 @@
+_target_: pytorch_lightning.Trainer
+devices: [0]
+accelerator: gpu
+min_epochs: 1
+max_epochs: 10
+log_every_n_steps: 10
+precision: 16
+strategy: auto
+num_nodes: 1
+fast_dev_run: false
+max_time:
+limit_train_batches:
+limit_val_batches:
+limit_test_batches:
+limit_predict_batches:
+overfit_batches: 0.0
+val_check_interval:
+check_val_every_n_epoch: 1
+num_sanity_val_steps:
+enable_checkpointing:
+enable_progress_bar:
+enable_model_summary:
+accumulate_grad_batches: 1
+gradient_clip_val:
+gradient_clip_algorithm:
+deterministic:
+benchmark:
+inference_mode: True
+use_distributed_sampler: True
+profiler:
+detect_anomaly: False
+barebones: False
+plugins:
+sync_batchnorm: False
+reload_dataloaders_every_n_epochs: 0
+default_root_dir:

quadra/configs/trainer/lightning_multigpu.yaml

@@ -0,0 +1,37 @@
+_target_: pytorch_lightning.Trainer
+devices: [0, 1]
+accelerator: gpu
+min_epochs: 1
+max_epochs: 10
+strategy:
+  _target_: pytorch_lightning.strategies.DDPStrategy
+  find_unused_parameters: false
+log_every_n_steps: 10
+num_nodes: 1
+fast_dev_run: false
+max_time:
+limit_train_batches:
+limit_val_batches:
+limit_test_batches:
+limit_predict_batches:
+overfit_batches: 0.0
+val_check_interval:
+check_val_every_n_epoch: 1
+num_sanity_val_steps:
+enable_checkpointing:
+enable_progress_bar:
+enable_model_summary:
+accumulate_grad_batches: 1
+gradient_clip_val:
+gradient_clip_algorithm:
+deterministic:
+benchmark:
+inference_mode: True
+use_distributed_sampler: True
+profiler:
+detect_anomaly: False
+barebones: False
+plugins:
+sync_batchnorm: False
+reload_dataloaders_every_n_epochs: 0
+default_root_dir:

quadra/configs/trainer/sklearn_classification.yaml

@@ -0,0 +1,7 @@
+_target_: quadra.trainers.classification.SklearnClassificationTrainer
+random_state: 42
+input_shape:
+  - ${transforms.input_height}
+  - ${transforms.input_width}
+  - 3
+iteration_over_training: 1

quadra/configs/transforms/byol.yaml

@@ -0,0 +1,47 @@
+defaults:
+  - default_resize
+  - _self_
+
+flip_and_jitter:
+  _target_: albumentations.Compose
+  transforms:
+    - _target_: albumentations.HorizontalFlip
+      p: 0.5
+    - _target_: albumentations.ColorJitter
+      brightness: 0.4
+      contrast: 0.4
+      saturation: 0.2
+      hue: 0.1
+      p: 0.8
+    - _target_: albumentations.ToGray
+      p: 0.2
+
+augmentation1:
+  _target_: albumentations.Compose
+  transforms:
+    - _target_: albumentations.RandomResizedCrop
+      height: ${transforms.input_height}
+      width: ${transforms.input_width}
+      scale: [0.08, 1.0]
+    - ${transforms.flip_and_jitter}
+    - _target_: albumentations.GaussianBlur
+      blur_limit: 23
+      sigma_limit: [0.1, 2]
+      p: 1.0
+    - ${transforms.normalize}
+
+augmentation2:
+  _target_: albumentations.Compose
+  transforms:
+    - _target_: albumentations.RandomResizedCrop
+      height: ${transforms.input_height}
+      width: ${transforms.input_width}
+      scale: [0.08, 1.0]
+    - ${transforms.flip_and_jitter}
+    - _target_: albumentations.GaussianBlur
+      blur_limit: 23
+      sigma_limit: [0.1, 2]
+      p: 0.1
+    - _target_: albumentations.Solarize
+      p: 0.2
+    - ${transforms.normalize}

quadra/configs/transforms/byol_no_random_resize.yaml

@@ -0,0 +1,61 @@
+defaults:
+  - default_resize
+  - _self_
+
+flip_and_jitter:
+  _target_: albumentations.Compose
+  transforms:
+    - _target_: albumentations.HorizontalFlip
+      p: 0.5
+    - _target_: albumentations.ColorJitter
+      brightness: 0.4
+      contrast: 0.4
+      saturation: 0.2
+      hue: 0.1
+      p: 0.8
+    - _target_: albumentations.ToGray
+      p: 0.2
+
+augmentation1:
+  _target_: albumentations.Compose
+  transforms:
+    - _target_: albumentations.Resize
+      height: ${transforms.input_height}
+      width: ${transforms.input_width}
+      interpolation: 2
+    - _target_: albumentations.ShiftScaleRotate
+      shift_limit_x: 0.7
+      shift_limit_y: 0.1
+      rotate_limit: 0
+      scale_limit: 0
+      border_mode: 3 # cv2.BORDER_WRAP
+      p: 0.3
+    - ${transforms.flip_and_jitter}
+    - _target_: albumentations.GaussianBlur
+      blur_limit: 23
+      sigma_limit: [0.1, 2]
+      p: 1.0
+    - ${transforms.normalize}
+
+augmentation2:
+  _target_: albumentations.Compose
+  transforms:
+    - _target_: albumentations.Resize
+      height: ${transforms.input_height}
+      width: ${transforms.input_width}
+      interpolation: 2
+    - _target_: albumentations.ShiftScaleRotate
+      shift_limit_x: 0.7
+      shift_limit_y: 0
+      rotate_limit: 0
+      scale_limit: 0
+      border_mode: 3 # cv2.BORDER_WRAP
+      p: 0.3
+    - ${transforms.flip_and_jitter}
+    - _target_: albumentations.GaussianBlur
+      blur_limit: 23
+      sigma_limit: [0.1, 2]
+      p: 0.1
+    - _target_: albumentations.Solarize
+      p: 0.2
+    - ${transforms.normalize}