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

quadra/configs/experiment/base/anomaly/fastflow.yaml

@@ -0,0 +1,46 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/anomaly
+  - override /model: anomalib/fastflow
+  - override /optimizer: null
+  - override /scheduler: null
+  - override /transforms: default_resize
+  - override /loss: null
+  - override /task: anomalib/fastflow
+  - override /backbone: null
+  - override /trainer: lightning_gpu
+  - override /callbacks: default_anomalib
+  - _self_
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  test_batch_size: 32
+  task: ${model.dataset.task}
+  category:
+
+callbacks:
+  early_stopping:
+    patience: 3
+    monitor: validation_image_F1
+
+print_config: false
+
+core:
+  tag: "run"
+  name: fastflow_${model.model.backbone}_${trainer.max_epochs}
+  test_after_training: true
+
+logger:
+  mlflow:
+    experiment_name:
+    run_name: ${core.name}
+
+trainer:
+  devices: [0]
+  gradient_clip_val: 0
+  gradient_clip_algorithm: norm
+  check_val_every_n_epoch: 1 # Don't validate before extracting features.
+  val_check_interval: 1.0 # Don't validate before extracting features.
+  max_epochs: 500

quadra/configs/experiment/base/anomaly/inference.yaml

@@ -0,0 +1,21 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/anomaly
+  - override /model: null
+  - override /optimizer: null
+  - override /scheduler: null
+  - override /transforms: default_resize
+  - override /loss: null
+  - override /task: anomalib/inference
+  - override /backbone: null
+  - override /trainer: null
+  - _self_
+
+datamodule:
+  phase: test
+
+task:
+  model_path: ???
+  use_training_threshold: false
+  training_threshold_type: image

quadra/configs/experiment/base/anomaly/padim.yaml

@@ -0,0 +1,37 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/anomaly
+  - override /model: anomalib/padim
+  - override /optimizer: null
+  - override /scheduler: null
+  - override /transforms: default_resize
+  - override /loss: null
+  - override /task: anomalib/padim
+  - override /backbone: null
+  - override /trainer: lightning_gpu
+  - override /callbacks: default_anomalib
+  - _self_
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  test_batch_size: 32
+  task: ${model.dataset.task}
+  category:
+  phase: train
+
+core:
+  tag: "run"
+  name: padim_${model.model.backbone}_${trainer.max_epochs}
+  test_after_training: true
+
+logger:
+  mlflow:
+    experiment_name:
+    run_name: ${core.name}
+
+trainer:
+  devices: [2]
+  check_val_every_n_epoch: ${trainer.max_epochs} # Don't validate before extracting features.
+  max_epochs: 1

quadra/configs/experiment/base/anomaly/patchcore.yaml

@@ -0,0 +1,37 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/anomaly
+  - override /model: anomalib/patchcore
+  - override /optimizer: null
+  - override /scheduler: null
+  - override /transforms: default_resize
+  - override /loss: null
+  - override /task: anomalib/patchcore
+  - override /backbone: null
+  - override /trainer: lightning_gpu
+  - override /callbacks: default_anomalib
+  - _self_
+
+datamodule:
+  num_workers: 12
+  train_batch_size: 32
+  test_batch_size: 32
+  task: ${model.dataset.task}
+  category:
+  phase: train
+
+core:
+  tag: "run"
+  name: patchcore_${model.model.backbone}_${trainer.max_epochs}
+  test_after_training: true
+
+logger:
+  mlflow:
+    experiment_name:
+    run_name: ${core.name}
+
+trainer:
+  devices: [2]
+  check_val_every_n_epoch: ${trainer.max_epochs} # Don't validate before extracting features.
+  max_epochs: 1

quadra/configs/experiment/base/classification/classification.yaml

@@ -0,0 +1,73 @@
+# @package _global_
+defaults:
+  - override /backbone: resnet18
+  - override /datamodule: base/classification
+  - override /loss: cross_entropy
+  - override /model: classification
+  - override /optimizer: adam
+  - override /task: classification
+  - override /scheduler: rop
+  - override /transforms: default_resize
+
+export:
+  types: [torchscript, pytorch]
+
+datamodule:
+  num_workers: 8
+  batch_size: 32
+  data_path: ???
+
+print_config: true
+
+model:
+  num_classes:
+  module:
+    lr_scheduler_interval: "epoch"
+
+task:
+  lr_multiplier: 0.0
+  run_test: True
+  report: True
+  gradcam: True
+  output:
+    example: True
+
+core:
+  tag: "run"
+  upload_artifacts: true
+  name: classification_base_${trainer.max_epochs}
+
+logger:
+  mlflow:
+    experiment_name: classification_base
+    run_name: ${core.name}
+
+backbone:
+  model:
+    pretrained: True
+    freeze: False
+  freeze_parameters_name:
+
+trainer:
+  max_epochs: 200
+  check_val_every_n_epoch: 1
+  log_every_n_steps: 1
+  devices: [0]
+
+scheduler:
+  patience: 20
+  factor: 0.9
+  verbose: False
+  threshold: 0.01
+
+callbacks:
+  early_stopping:
+    _target_: pytorch_lightning.callbacks.EarlyStopping
+    monitor: val_loss_epoch
+    min_delta: 0.01
+    mode: min
+    patience: 35
+    verbose: false
+    stopping_threshold: 0
+  model_checkpoint:
+    monitor: val_loss_epoch

quadra/configs/experiment/base/classification/classification_evaluation.yaml

@@ -0,0 +1,25 @@
+# @package _global_
+defaults:
+  - override /datamodule: base/classification
+  - override /transforms: default_resize
+  - override /task: classification_evaluation
+
+datamodule:
+  num_workers: 6
+  batch_size: 32
+
+core:
+  tag: "run"
+  upload_artifacts: true
+  name: classification_evalutation_base
+
+logger:
+  mlflow:
+    experiment_name: classification_evalutation_base
+    run_name: ${core.name}
+
+task:
+  gradcam: true
+  output:
+    example: true
+  model_path: ???

quadra/configs/experiment/base/classification/multilabel_classification.yaml

@@ -0,0 +1,41 @@
+# @package _global_
+defaults:
+  - override /backbone: resnet18
+  - override /datamodule: base/multilabel_classification
+  - override /loss: asl
+  - override /model: multilabel_classification
+  - override /optimizer: adam
+  - override /task: classification
+  - override /scheduler: rop
+  - override /transforms: default_resize
+
+export:
+  types: [torchscript, pytorch]
+
+datamodule:
+  num_workers: 8
+  batch_size: 32
+  test_batch_size: 32
+
+print_config: true
+
+core:
+  tag: "run"
+  test_after_training: true
+  upload_artifacts: true
+  name: multilabel-classification
+
+logger:
+  mlflow:
+    experiment_name: ${core.name}
+    run_name: ${core.name}
+
+backbone:
+  model:
+    pretrained: True
+    freeze: False
+
+trainer:
+  devices: [0]
+  max_epochs: 300
+  check_val_every_n_epoch: 1

quadra/configs/experiment/base/classification/sklearn_classification.yaml

@@ -0,0 +1,27 @@
+# @package _global_
+
+defaults:
+  - override /model: logistic_regression
+  - override /transforms: default_resize
+  - override /task: sklearn_classification
+  - override /backbone: dino_vitb8
+  - override /trainer: sklearn_classification
+  - override /datamodule: base/sklearn_classification
+
+export:
+  types: [torchscript, pytorch]
+
+backbone:
+  model:
+    pretrained: true
+    freeze: true
+
+core:
+  tag: "run"
+  name: "sklearn-classification"
+
+datamodule:
+  num_workers: 8
+  batch_size: 32
+  phase: train
+  n_splits: 1

quadra/configs/experiment/base/classification/sklearn_classification_patch.yaml

@@ -0,0 +1,25 @@
+# @package _global_
+
+defaults:
+  - override /model: logistic_regression
+  - override /transforms: default_resize
+  - override /task: sklearn_classification_patch
+  - override /backbone: dino_vitb8
+  - override /trainer: sklearn_classification
+  - override /datamodule: base/sklearn_classification_patch
+
+backbone:
+  model:
+    pretrained: true
+    freeze: true
+
+core:
+  tag: "run"
+  name: "sklearn-classification-patch"
+
+trainer:
+  iteration_over_training: 20 # Regulate how many patches are extracted
+
+datamodule:
+  num_workers: 8
+  batch_size: 32

quadra/configs/experiment/base/classification/sklearn_classification_patch_test.yaml

@@ -0,0 +1,18 @@
+# @package _global_
+
+defaults:
+  - override /transforms: default_resize
+  - override /task: sklearn_classification_patch_test
+  - override /datamodule: base/sklearn_classification_patch
+  - override /trainer: sklearn_classification
+
+core:
+  tag: "run"
+  name: "sklearn-classification-patch-test"
+
+task:
+  model_path: ???
+
+datamodule:
+  num_workers: 8
+  batch_size: 32

quadra/configs/experiment/base/classification/sklearn_classification_test.yaml

@@ -0,0 +1,25 @@
+# @package _global_
+
+defaults:
+  - override /transforms: default_resize
+  - override /task: sklearn_classification_test
+  - override /trainer: sklearn_classification
+  - override /datamodule: base/sklearn_classification
+
+core:
+  tag: run
+  name: sklearn-classification-test
+
+task:
+  device: cuda:0
+  gradcam: false
+  output:
+    folder: classification_experiment
+    report: true
+    example: true
+  model_path: ???
+
+datamodule:
+  num_workers: 8
+  batch_size: 32
+  phase: test

quadra/configs/experiment/base/segmentation/smp.yaml

@@ -0,0 +1,30 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/segmentation
+  - override /model: smp
+  - override /optimizer: adam
+  - override /scheduler: rop
+  - override /transforms: default_resize
+  - override /loss: smp_dice
+  - override /task: segmentation
+  - override /backbone: smp
+  - override /trainer: lightning_gpu
+core:
+  tag: "run"
+  name: "quadra_default"
+  upload_artifacts: True
+
+trainer:
+  devices: [0]
+  max_epochs: 100
+  num_sanity_val_steps: 0
+
+datamodule:
+  num_workers: 5
+  batch_size: 32
+
+task:
+  report: true
+  evaluate:
+    analysis: true

quadra/configs/experiment/base/segmentation/smp_evaluation.yaml

@@ -0,0 +1,17 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/segmentation
+  - override /transforms: default_resize
+  - override /task: segmentation_evaluation
+core:
+  tag: "run"
+  name: "quadra_default"
+  upload_artifacts: True
+
+task:
+  model_path: ???
+
+datamodule:
+  num_workers: 5
+  batch_size: 32

quadra/configs/experiment/base/segmentation/smp_multiclass.yaml

@@ -0,0 +1,26 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/segmentation_multiclass
+  - override /model: smp_multiclass
+  - override /optimizer: adam
+  - override /scheduler: rop
+  - override /transforms: default_resize
+  - override /loss: smp_dice_multiclass
+  - override /task: segmentation
+  - override /backbone: smp
+  - override /trainer: lightning_gpu
+
+trainer:
+  devices: [0]
+  max_epochs: 100
+  num_sanity_val_steps: 0
+
+core:
+  tag: "run"
+  name: "quadra_default"
+  upload_artifacts: True
+
+datamodule:
+  num_workers: 8
+  batch_size: 32

quadra/configs/experiment/base/segmentation/smp_multiclass_evaluation.yaml

@@ -0,0 +1,18 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/segmentation_multiclass
+  - override /transforms: default_resize
+  - override /task: segmentation_evaluation
+
+core:
+  tag: "run"
+  name: "quadra_default"
+  upload_artifacts: True
+
+task:
+  model_path: ???
+
+datamodule:
+  num_workers: 8
+  batch_size: 32

quadra/configs/experiment/base/ssl/barlow.yaml

@@ -0,0 +1,48 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/ssl
+  - override /backbone: resnet18
+  - override /model: barlow
+  - override /optimizer: lars
+  - override /scheduler: warmup
+  - override /transforms: byol
+  - override /loss: barlow
+  - override /task: ssl
+  - override /trainer: lightning_gpu_fp16
+core:
+  tag: "run"
+  name: "barlow_ssl"
+task:
+  _target_: quadra.tasks.ssl.Barlow
+
+trainer:
+  devices: [0]
+  max_epochs: 300
+  num_sanity_val_steps: 0
+  gradient_clip_val: 2.0
+  check_val_every_n_epoch: 10
+  gradient_clip_algorithm: "value"
+
+datamodule:
+  num_workers: 16
+  batch_size: 128
+  augmentation_dataset:
+    _target_: quadra.datasets.TwoAugmentationDataset
+    transform:
+      - ${transforms.augmentation1}
+      - ${transforms.augmentation2}
+    dataset: null
+callbacks:
+  model_checkpoint:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: "val_acc"
+    mode: "max"
+
+optimizer:
+  lr: 0.0
+
+scheduler:
+  init_lr:
+    - 0.001
+  linear_warmup_epochs: 10

quadra/configs/experiment/base/ssl/byol.yaml

@@ -0,0 +1,43 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/ssl
+  - override /backbone: resnet18
+  - override /model: byol
+  - override /optimizer: lars
+  - override /scheduler: warmup
+  - override /transforms: byol
+  - override /loss: byol
+  - override /task: ssl
+  - override /trainer: lightning_gpu_fp16
+core:
+  tag: "run"
+  name: "byol_ssl"
+task:
+  _target_: quadra.tasks.ssl.BYOL
+
+callbacks:
+  model_checkpoint:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: "val_acc"
+    mode: "max"
+
+trainer:
+  devices: [0]
+  max_epochs: 500
+  num_sanity_val_steps: 0
+  check_val_every_n_epoch: 10
+
+datamodule:
+  num_workers: 12
+  batch_size: 256
+  augmentation_dataset:
+    _target_: quadra.datasets.TwoAugmentationDataset
+    transform:
+      - ${transforms.augmentation1}
+      - ${transforms.augmentation2}
+    dataset: null
+
+scheduler:
+  init_lr:
+    - 0.4

quadra/configs/experiment/base/ssl/dino.yaml

@@ -0,0 +1,46 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/ssl
+  - override /backbone: resnet18
+  - override /model: dino
+  - override /optimizer: adamw
+  - override /scheduler: warmup
+  - override /transforms: dino
+  - override /loss: dino
+  - override /task: ssl
+  - override /trainer: lightning_gpu_fp16
+
+core:
+  tag: "run"
+  name: "dino_ssl"
+task:
+  _target_: quadra.tasks.ssl.DINO
+
+callbacks:
+  model_checkpoint:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: "val_acc"
+    mode: "max"
+
+trainer:
+  devices: [0]
+  max_epochs: 500
+  num_sanity_val_steps: 0
+  gradient_clip_val: 3.0
+  check_val_every_n_epoch: 10
+  gradient_clip_algorithm: "value"
+
+datamodule:
+  num_workers: 6
+  batch_size: 128
+  augmentation_dataset:
+    _target_: quadra.datasets.TwoSetAugmentationDataset
+    global_transforms: ${transforms.global_transforms}
+    local_transform: ${transforms.local_transform}
+    num_local_transforms: 10
+    dataset: null
+
+scheduler:
+  init_lr:
+    - 0.0001

quadra/configs/experiment/base/ssl/linear_eval.yaml

@@ -0,0 +1,52 @@
+# @package _global_
+
+defaults:
+  - override /datamodule: base/ssl
+  - override /model: classification
+  - override /optimizer: sgd
+  - override /scheduler: warmup
+  - override /transforms: linear_eval
+  - override /task: classification
+  - override /backbone: resnet18
+  - override /trainer: lightning_multigpu
+ssl_task: ???
+
+core:
+  tag: "run"
+  name: linear-eval-${ssl_task}
+
+backbone:
+  model:
+    num_classes: 2
+
+trainer:
+  devices: [2, 3]
+  max_epochs: 90
+  num_sanity_val_steps: 0
+  precision: 16
+  check_val_every_n_epoch: 10
+  sync_batchnorm: true
+
+datamodule:
+  num_workers: 8
+  batch_size: 1024
+  use_concordant: false
+  use_discordant: false
+
+callbacks:
+  model_checkpoint:
+    _target_: pytorch_lightning.callbacks.ModelCheckpoint
+    monitor: "val_acc"
+    mode: "max"
+
+optimizer:
+  lr: 0.0
+
+scheduler:
+  init_lr:
+    - 0.001
+  linear_warmup_epochs: 0
+
+logger:
+  mlflow:
+    experiment_name: ${core.name}