Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Generalize SSL functionality to work on other datasets #555

Merged
merged 25 commits into from
Sep 15, 2021
Merged
Changes from 1 commit
Commits
Show all changes
25 commits
Select commit Hold shift + click to select a range
42072a6
extending _get_transforms to accept new datasets
vale-salvatelli Aug 24, 2021
a8ebe13
expand get_cxr_ssl_transform to avoid hidden channel expansion
vale-salvatelli Aug 24, 2021
c2c5fe7
drop_last set as parameter of InnerEyeVisionDataModule
vale-salvatelli Aug 24, 2021
682d2ab
drop_last is now a SSLContainer parameter
vale-salvatelli Aug 24, 2021
68cb45c
Updating Changelog
vale-salvatelli Aug 24, 2021
bdf4ca6
Fix PEP8
vale-salvatelli Aug 24, 2021
fcf27ed
fixing mypy error
vale-salvatelli Aug 25, 2021
bccdb6b
still one fix
vale-salvatelli Aug 25, 2021
68ae373
Merge branch 'main' into vsalva/generalize_ssl
vale-salvatelli Aug 25, 2021
26522c8
Updating to main
vale-salvatelli Aug 25, 2021
68dd10c
generalize function names for readibility
vale-salvatelli Aug 26, 2021
d72a36b
Updating documentation
vale-salvatelli Aug 26, 2021
daeaab1
Updating documentation
vale-salvatelli Aug 26, 2021
6509e4f
removing unexpected changes in amlignore
vale-salvatelli Aug 26, 2021
bc5a81c
Adding test
vale-salvatelli Aug 26, 2021
fc22df9
Adding bits to the test
vale-salvatelli Aug 26, 2021
d74eaf4
committing to switch branch, test_transform pipeline still to be fixed
vale-salvatelli Sep 1, 2021
0cc7893
fixing test
vale-salvatelli Sep 14, 2021
c474713
remove TODO
vale-salvatelli Sep 14, 2021
7cf4459
fixing conlicts
vale-salvatelli Sep 14, 2021
9cda074
fixing flake8
vale-salvatelli Sep 14, 2021
7fa0dbd
fixing flake8 for real
vale-salvatelli Sep 14, 2021
1b978dd
fixing more flake8
vale-salvatelli Sep 14, 2021
7af305c
docstring changed
vale-salvatelli Sep 15, 2021
0c255a5
docstring changed, thanks Mel
vale-salvatelli Sep 15, 2021
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
docstring changed, thanks Mel
  • Loading branch information
vale-salvatelli committed Sep 15, 2021
commit 0c255a577b64ab91653a9ab0463f067310845dcd
2 changes: 1 addition & 1 deletion InnerEye/ML/SSL/lightning_containers/ssl_container.py
Original file line number Diff line number Diff line change
Expand Up @@ -226,7 +226,7 @@ def _get_transforms(self, augmentation_config: Optional[CfgNode],
:param dataset_name: name of the dataset, value has to be in SSLDatasetName, determines which transformation
pipeline to return.
:param is_ssl_encoder_module: if True the transformation pipeline will yield two versions of the image it is
vale-salvatelli marked this conversation as resolved.
Show resolved Hide resolved
applied on and it applies the same transformations for training and validation. Note that if your transformation
applied on and it applies the training transformations also at validation time. Note that if your transformation
does not contain any randomness, the pipeline will return two identical copies. If False, it will return only one
transformation.
:return: training transformation pipeline and validation transformation pipeline.
Expand Down